frozenleaves/pytorch
1
0
Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2025-10-20 12:54:11 +08:00
Code Issues Packages Projects Releases Wiki Activity

[torchfuzz] remove unneeded try catch (#163557)

Browse Source 
Pull Request resolved: https://github.com/pytorch/pytorch/pull/163557
Approved by: https://github.com/laithsakka
ghstack dependencies: #163547, #163553, #163554, #163555, #163556
This commit is contained in:
bobrenjc93
2025-09-22 15:14:38 -07:00
committed by PyTorch MergeBot
parent 1545bb1c00
commit 309fe03f4b
1 changed files with 5 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
tools/experimental/dynamic_shapes/torchfuzz/ops_fuzzer.py
View File

@ -246,19 +246,13 @@ def fuzz_op(target_spec: Spec, depth, stack_size) -> tuple[str, list[Spec]]:
random.choice(all_ops) if all_ops else ("arg", get_operator("arg"))
)
# Use the operator to decompose the target spec into input specs
try:
if chosen_operator is None:
# If no operator found, fallback to arg
return _get_arg_args_specs(target_spec)
input_specs = chosen_operator.fuzz_inputs_specs(target_spec)
return chosen_op_name, input_specs
except Exception as e:
# Fallback to arg if decomposition fails
print(f"Warning: operator {chosen_op_name} decomposition failed: {e}")
if chosen_operator is None:
# If no operator found, fallback to arg
return _get_arg_args_specs(target_spec)
input_specs = chosen_operator.fuzz_inputs_specs(target_spec)
return chosen_op_name, input_specs
# Global counter for generating unique argument IDs
_next_arg_id = 0