From 3e9bacdb514af36dea80f4e40d251f2a2cca4c4e Mon Sep 17 00:00:00 2001
From: Aart Bik <39774503+aartbik@users.noreply.github.com>
Date: Wed, 3 Jan 2024 16:10:50 -0800
Subject: [PATCH] [torch-mlir] update e2e test class documentation (#2722)

The doc seems copy-and-paste from the linalg-on-tensors class
---
 .../python/torch_mlir_e2e_test/configs/stablehlo_backend.py   | 4 ++--
 .../pt1/python/torch_mlir_e2e_test/configs/tosa_backend.py    | 4 ++--
 2 files changed, 4 insertions(+), 4 deletions(-)

diff --git a/projects/pt1/python/torch_mlir_e2e_test/configs/stablehlo_backend.py b/projects/pt1/python/torch_mlir_e2e_test/configs/stablehlo_backend.py
index 45f32bb0b3fe..8a244b756e6c 100644
--- a/projects/pt1/python/torch_mlir_e2e_test/configs/stablehlo_backend.py
+++ b/projects/pt1/python/torch_mlir_e2e_test/configs/stablehlo_backend.py
@@ -18,10 +18,10 @@
 
 
 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):
diff --git a/projects/pt1/python/torch_mlir_e2e_test/configs/tosa_backend.py b/projects/pt1/python/torch_mlir_e2e_test/configs/tosa_backend.py
index 89b90567b1d4..8efab87a2bfe 100644
--- a/projects/pt1/python/torch_mlir_e2e_test/configs/tosa_backend.py
+++ b/projects/pt1/python/torch_mlir_e2e_test/configs/tosa_backend.py
@@ -18,10 +18,10 @@
 
 
 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__()