Skip to content

Commit 3e9bacd

Browse files
authored
[torch-mlir] update e2e test class documentation (#2722)
The doc seems copy-and-paste from the linalg-on-tensors class
1 parent 1778314 commit 3e9bacd

File tree

2 files changed

+4
-4
lines changed

2 files changed

+4
-4
lines changed

projects/pt1/python/torch_mlir_e2e_test/configs/stablehlo_backend.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -18,10 +18,10 @@
1818

1919

2020
class StablehloBackendTestConfig(TestConfig):
21-
"""Base class for TestConfig's that are implemented with linalg-on-tensors.
21+
"""Base class for TestConfig's that are implemented with StableHLO.
2222
2323
This class handles all the common lowering that torch-mlir does before
24-
reaching the linalg-on-tensors abstraction level.
24+
reaching the StableHLO abstraction level.
2525
"""
2626

2727
def __init__(self, backend: StablehloBackend):

projects/pt1/python/torch_mlir_e2e_test/configs/tosa_backend.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -18,10 +18,10 @@
1818

1919

2020
class TosaBackendTestConfig(TestConfig):
21-
"""Base class for TestConfig's that are implemented with linalg-on-tensors.
21+
"""Base class for TestConfig's that are implemented with TOSA.
2222
2323
This class handles all the common lowering that torch-mlir does before
24-
reaching the linalg-on-tensors abstraction level.
24+
reaching the TOSA abstraction level.
2525
"""
2626
def __init__(self, backend: TosaBackend, use_make_fx: bool = False):
2727
super().__init__()

0 commit comments

Comments
 (0)