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", ] 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":