Skip to content

Commit a7ad039

Browse files
Fix for API drift
1 parent e30dea5 commit a7ad039

File tree

1 file changed

+1
-1
lines changed
  • python/shark_turbine/runtime/op_reg

1 file changed

+1
-1
lines changed

python/shark_turbine/runtime/op_reg/base.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -618,7 +618,7 @@ def yield_results(self, *results: Value):
618618

619619
def constant_index(self, i: int) -> Value:
620620
"""Builds a constant index value."""
621-
return arith_d.constant(IntegerAttr.get(IndexType.get(), i))
621+
return arith_d.constant(IndexType.get(), IntegerAttr.get(IndexType.get(), i))
622622

623623

624624
class FreeFuncKernelBuilder(KernelBuilder):

0 commit comments

Comments
 (0)