Skip to content

Commit 38561ca

Browse files
committed
Dependency changes seemed to have added int64 as valid dtype
1 parent 8394892 commit 38561ca

File tree

1 file changed

+2
-2
lines changed
  • tripy/nvtripy/frontend/ops/binary

1 file changed

+2
-2
lines changed

tripy/nvtripy/frontend/ops/binary/pow.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -22,7 +22,7 @@
2222
@register_tensor_method("__pow__")
2323
@wrappers.interface(
2424
dtype_constraints={"self": "T1", "other": "T1", wrappers.RETURN_VALUE: "T1"},
25-
dtype_variables={"T1": ["float32", "float16", "bfloat16", "int8"]},
25+
dtype_variables={"T1": ["float32", "float16", "bfloat16", "int8", "int64"]},
2626
convert_to_tensors=True,
2727
)
2828
def __pow__(self: "nvtripy.Tensor", other: TensorLike) -> "nvtripy.Tensor":
@@ -52,7 +52,7 @@ def __pow__(self: "nvtripy.Tensor", other: TensorLike) -> "nvtripy.Tensor":
5252
@register_tensor_method("__rpow__")
5353
@wrappers.interface(
5454
dtype_constraints={"self": "T1", "other": "T1", wrappers.RETURN_VALUE: "T1"},
55-
dtype_variables={"T1": ["float32", "float16", "bfloat16", "int8"]},
55+
dtype_variables={"T1": ["float32", "float16", "bfloat16", "int8", "int64"]},
5656
convert_to_tensors=True,
5757
)
5858
def __rpow__(self: "nvtripy.Tensor", other: TensorLike) -> "nvtripy.Tensor":

0 commit comments

Comments
 (0)