Skip to content

Commit

Permalink
add is_integer to support example_value function whitelist (pytorch#1…
Browse files Browse the repository at this point in the history
…39484)

Fixes python test/dynamo/test_dynamic_shapes.py DynamicShapesFunctionTests.test_is_integer_dynamic_shapes when specialize_float=False

Pull Request resolved: pytorch#139484
Approved by: https://github.com/ezyang
ghstack dependencies: pytorch#139451, pytorch#139482
  • Loading branch information
bobrenjc93 authored and rahulsingh-intel committed Nov 5, 2024
1 parent ba16b46 commit 42cc947
Showing 1 changed file with 1 addition and 0 deletions.
1 change: 1 addition & 0 deletions torch/_dynamo/variables/builder.py
Original file line number Diff line number Diff line change
Expand Up @@ -2373,6 +2373,7 @@ def handle_traced_output(example_value, tx, proxy, options, subclass_type, targe
torch.backends.cuda.is_flash_attention_available,
torch.backends.cuda.can_use_flash_attention,
torch.backends.cuda.can_use_efficient_attention,
"is_integer",
]
+ list(supported_const_comparison_op_values.keys())
):
Expand Down

0 comments on commit 42cc947

Please sign in to comment.