Add TTOPInterface funcs for supportsCPUExecution + supportsOnlyCPUExe…

…cution (#1310)
tenstorrent · Nov 18, 2024 · 82ffe2b · 82ffe2b
1 parent 4ed0821
commit 82ffe2b
Showing 1 changed file with 20 additions and 0 deletions.
diff --git a/include/ttmlir/Dialect/TTIR/IR/TTIROpsInterfaces.td b/include/ttmlir/Dialect/TTIR/IR/TTIROpsInterfaces.td
@@ -41,6 +41,26 @@ def TTIROpInterface : OpInterface<"TTIROp"> {
       /*methodBody=*/"",
       /*defaultImplementation=*/"return ::mlir::tt::getCurrentScopeDevice($_op);"
     >,
+    InterfaceMethod<
+      /*desc=*/[{
+        Return whether op can be run on CPU.
+      }],
+      /*retTy=*/"bool",
+      /*methodName=*/"supportsCPUExecution",
+      /*args=*/(ins),
+      /*methodBody=*/"",
+      /*defaultImplementation=*/"return false;"
+    >,
+    InterfaceMethod<
+      /*desc=*/[{
+        Return whether op must be run on CPU.
+      }],
+      /*retTy=*/"bool",
+      /*methodName=*/"supportsOnlyCPUExecution",
+      /*args=*/(ins),
+      /*methodBody=*/"",
+      /*defaultImplementation=*/"return false;"
+    >,
   ];
 }