From 504de3fb90547e82c45f277cacecf2f99b284305 Mon Sep 17 00:00:00 2001 From: Parth Chadha Date: Fri, 6 Dec 2024 16:35:15 -0800 Subject: [PATCH 1/2] Update mlir-tensorrt version to 0.1.38 --- tripy/pyproject.toml | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tripy/pyproject.toml b/tripy/pyproject.toml index 140227828..c5d4ef397 100644 --- 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", ] From 9ea5a4fd78c5950ae4c9be518f8c500c4cedeb88 Mon Sep 17 00:00:00 2001 From: Parth Chadha Date: Mon, 9 Dec 2024 10:52:55 -0800 Subject: [PATCH 2/2] Remove float8 from failing tests --- tripy/tripy/frontend/ops/stack.py | 2 +- tripy/tripy/frontend/trace/ops/binary_elementwise.py | 4 ++-- tripy/tripy/frontend/trace/ops/concatenate.py | 2 +- 3 files changed, 4 insertions(+), 4 deletions(-) diff --git a/tripy/tripy/frontend/ops/stack.py b/tripy/tripy/frontend/ops/stack.py index 5b64e4094..3a45dd5a7 100644 --- 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 index f017ad855..bf1cb03a1 100644 --- 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 index 8ba597fea..b4dc1d9e3 100644 --- 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":