pytorch
69969e52 - Cast computation_node_input_size to int (#103677)

Commit
2 years ago
Cast computation_node_input_size to int (#103677) This bandaid fixes yolov3 with automatic_dynamic_shapes. A more proper fix probably is to figure out why when we have ``` TypeError: mkldnn_reorder_conv2d_weight(): argument 'input_size' (position 6) must be tuple of ints, but found element of type SymInt at pos 1 ``` where the SymInt is known to be constant, we aren't willing to coerce it to int. Signed-off-by: Edward Z. Yang <ezyang@meta.com> Pull Request resolved: https://github.com/pytorch/pytorch/pull/103677 Approved by: https://github.com/voznesenskym
Author
Committer
Parents
Loading