Update mlir-tensorrt version to 0.1.38 (#431)

NVIDIA · Dec 9, 2024 · bc87585 · bc87585
1 parent 47c11d5
commit bc87585
Show file tree

Hide file tree

Showing 4 changed files with 6 additions and 6 deletions.
diff --git a/tripy/pyproject.toml b/tripy/pyproject.toml
@@ -8,8 +8,8 @@ requires-python = ">= 3.9"
 license = {text = "Apache 2.0"}
 dependencies = [
   "tensorrt~=10.0",
-  "mlir-tensorrt-compiler==0.1.37+cuda12.trt102",
-  "mlir-tensorrt-runtime==0.1.37+cuda12.trt102",
+  "mlir-tensorrt-compiler==0.1.38+cuda12.trt105",
+  "mlir-tensorrt-runtime==0.1.38+cuda12.trt105",
   "colored==2.2.3",
 ]
 

diff --git a/tripy/tripy/frontend/ops/stack.py b/tripy/tripy/frontend/ops/stack.py
@@ -23,7 +23,7 @@
 @constraints.dtypes(
     constraints={"tensors": "T1", constraints.RETURN_VALUE: "T1"},
     variables={
-        "T1": ["float32", "float16", "bfloat16", "float8", "int4", "int8", "int32", "int64", "bool"],
+        "T1": ["float32", "float16", "bfloat16", "int4", "int8", "int32", "int64", "bool"],
     },
 )
 def stack(tensors: Sequence["tripy.Tensor"], dim: int = 0) -> "tripy.Tensor":

diff --git a/tripy/tripy/frontend/trace/ops/binary_elementwise.py b/tripy/tripy/frontend/trace/ops/binary_elementwise.py
@@ -512,7 +512,7 @@ def __rfloordiv__(self: "tripy.Tensor", other: TensorLike) -> "tripy.Tensor":
 @frontend_utils.convert_to_tensors()
 @constraints.dtypes(
     constraints={"self": "T1", "other": "T1", constraints.RETURN_VALUE: "T1"},
-    variables={"T1": ["float32", "float16", "bfloat16", "float8"]},
+    variables={"T1": ["float32", "float16", "bfloat16"]},
 )
 def __mod__(self: "tripy.Tensor", other: TensorLike) -> "tripy.Tensor":
     """
@@ -543,7 +543,7 @@ def __mod__(self: "tripy.Tensor", other: TensorLike) -> "tripy.Tensor":
 @frontend_utils.convert_to_tensors()
 @constraints.dtypes(
     constraints={"self": "T1", "other": "T1", constraints.RETURN_VALUE: "T1"},
-    variables={"T1": ["float32", "float16", "bfloat16", "float8"]},
+    variables={"T1": ["float32", "float16", "bfloat16"]},
 )
 def __rmod__(self: "tripy.Tensor", other: TensorLike) -> "tripy.Tensor":
     """

diff --git a/tripy/tripy/frontend/trace/ops/concatenate.py b/tripy/tripy/frontend/trace/ops/concatenate.py
@@ -45,7 +45,7 @@ def to_flat_ir(self, inputs, outputs):
 @constraints.dtypes(
     constraints={"tensors": "T1", constraints.RETURN_VALUE: "T1"},
     variables={
-        "T1": ["float32", "float16", "bfloat16", "float8", "int4", "int8", "int32", "int64", "bool"],
+        "T1": ["float32", "float16", "bfloat16", "int4", "int8", "int32", "int64", "bool"],
     },
 )
 def concatenate(tensors: Sequence["tripy.Tensor"], dim: int) -> "tripy.Tensor":