llvm
/
torch-mlir
mirror of https://github.com/llvm/torch-mlir
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

[torch-mlir] update e2e test class documentation (#2722) 

The doc seems copy-and-paste from the linalg-on-tensors class
pull/2724/head
Aart Bik 2024-01-03 16:10:50 -08:00 committed by GitHub
parent 1778314620
commit 3e9bacdb51
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
projects/pt1/python/torch_mlir_e2e_test/configs/stablehlo_backend.py
View File

@ -18,10 +18,10 @@ from .utils import (
class StablehloBackendTestConfig(TestConfig):
"""Base class for TestConfig's that are implemented with linalg-on-tensors.
"""Base class for TestConfig's that are implemented with StableHLO.
This class handles all the common lowering that torch-mlir does before
reaching the linalg-on-tensors abstraction level.
reaching the StableHLO abstraction level.
"""
def __init__(self, backend: StablehloBackend):

4
projects/pt1/python/torch_mlir_e2e_test/configs/tosa_backend.py
View File

@ -18,10 +18,10 @@ from .utils import (
class TosaBackendTestConfig(TestConfig):
"""Base class for TestConfig's that are implemented with linalg-on-tensors.
"""Base class for TestConfig's that are implemented with TOSA.
This class handles all the common lowering that torch-mlir does before
reaching the linalg-on-tensors abstraction level.
reaching the TOSA abstraction level.
"""
def __init__(self, backend: TosaBackend, use_make_fx: bool = False):
super().__init__()