Rust: add constructor i64 IntImm

apache · Nov 9, 2023 · 2b598a9 · 2b598a9
1 parent ad978bc
commit 2b598a9
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/rust/tvm/src/ir/tir.rs b/rust/tvm/src/ir/tir.rs
@@ -56,6 +56,12 @@ impl From<i32> for IntImm {
     }
 }
 
+impl From<i64> for IntImm {
+    fn from(i : i64) -> IntImm {
+        IntImm::new(DataType::int(64, 1), i)
+    }
+}
+
 impl From<i32> for PrimExpr {
     fn from(i: i32) -> PrimExpr {
         IntImm::from(i).upcast()