Fix for API drift

nod-ai · Jan 31, 2024 · a7ad039 · a7ad039
1 parent e30dea5
commit a7ad039
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/python/shark_turbine/runtime/op_reg/base.py b/python/shark_turbine/runtime/op_reg/base.py
@@ -618,7 +618,7 @@ def yield_results(self, *results: Value):
 
     def constant_index(self, i: int) -> Value:
         """Builds a constant index value."""
-        return arith_d.constant(IntegerAttr.get(IndexType.get(), i))
+        return arith_d.constant(IndexType.get(), IntegerAttr.get(IndexType.get(), i))
 
 
 class FreeFuncKernelBuilder(KernelBuilder):