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

[torchfuzz] raise if Operator abstract method is not implemented (#164211)

Browse Source 
Pull Request resolved: https://github.com/pytorch/pytorch/pull/164211
Approved by: https://github.com/pianpwk
ghstack dependencies: #164034, #164209
This commit is contained in:
bobrenjc93
2025-09-30 14:02:58 -07:00
committed by PyTorch MergeBot
parent abfcce58a4
commit 1f3995cdc8
1 changed files with 3 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
tools/experimental/dynamic_shapes/torchfuzz/operators/base.py
View File

@ -23,10 +23,12 @@ class Operator(ABC):
Optional[str]: The torch operation name (e.g., "torch.ops.aten.add", "torch.nonzero").
Returns None for non-torch operations like "arg" and "constant".
"""
raise NotImplementedError("Subclasses must implement torch_op_name")
@abstractmethod
def can_produce(self, output_spec: Spec) -> bool:
"""Check if this operator can produce the given output spec."""
raise NotImplementedError("Subclasses must implement can_produce()")
def fuzz_inputs_specs(self, output_spec: Spec) -> list[Spec]:
"""
@ -40,6 +42,7 @@ class Operator(ABC):
self, output_name: str, input_names: list[str], output_spec: Spec
) -> str:
"""Generate code for this operation."""
raise NotImplementedError("Subclasses must implement codegen()")
def __str__(self) -> str:
"""String representation of the operator."""