Remove plugin WAR due to mlir-tensorrt issue #915 (#438)

NVIDIA · Dec 11, 2024 · 498eb8a · 498eb8a
1 parent 85d1d7d
commit 498eb8a
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 4 deletions.
diff --git a/tripy/tests/integration/test_plugin.py b/tripy/tests/integration/test_plugin.py
@@ -22,8 +22,7 @@
 
 class TestPlugin:
     def test_gelu(self):
-        # TODO: We add `+ 1` as a hack to work around MLIR-TRT Issue #915. We should be able to remove it once fixed
-        inp = tp.iota((2, 2)) + 1
+        inp = tp.iota((2, 2))
         out = tp.plugin(
             "CustomGeluPluginDynamic",
             [inp],

diff --git a/tripy/tripy/frontend/trace/ops/plugin.py b/tripy/tripy/frontend/trace/ops/plugin.py
@@ -75,8 +75,7 @@ def plugin(
         :linenos:
         :caption: Example
 
-        # TODO: We add `+ 1` as a hack to work around MLIR-TRT Issue #915. We should be able to remove it once fixed # doc: omit
-        inp = tp.iota((2, 1, 4)) + 1
+        inp = tp.iota((2, 1, 4))
         out = tp.plugin(
             "CustomGeluPluginDynamic",
             [inp],