mirror of
https://github.com/pytorch/pytorch.git
synced 2025-10-20 12:54:11 +08:00
c965d6dbb2
Pull Request resolved: https://github.com/pytorch/pytorch/pull/164688 Approved by: https://github.com/pianpwk ghstack dependencies: #164432, #164434, #164514, #164646, #164647, #164649, #164687
20 lines
522 B
Python
20 lines
522 B
Python
"""Torchfuzz package for generating and testing random PyTorch operations."""
|
|
|
|
# Make key classes available at package level
|
|
from .operators import get_operator, list_operators, register_operator
|
|
from .ops_fuzzer import fuzz_operation_graph, fuzz_spec, OperationGraph
|
|
from .tensor_fuzzer import ScalarSpec, Spec, TensorSpec
|
|
|
|
|
|
__all__ = [
|
|
"TensorSpec",
|
|
"ScalarSpec",
|
|
"Spec",
|
|
"OperationGraph",
|
|
"fuzz_operation_graph",
|
|
"fuzz_spec",
|
|
"get_operator",
|
|
"register_operator",
|
|
"list_operators",
|
|
]
|