[Transform] unsupported_dtype_legalize.cc - Only check cuda compute version for fp8 support on cuda target

Author: csullivan

src/tir/transforms/unsupported_dtype_legalize.cc

@@ -695,11 +695,13 @@ namespace transform {
 
 bool CheckDataTypeSupport(const Target& target, const std::string& support_func_name) {
   bool has_native_support = false;
-  if (const PackedFunc* get_cv =
-          tvm::runtime::Registry::Get("tvm.contrib.nvcc.get_compute_version")) {
-    std::string compute_version = (*get_cv)(target);
-    if (const PackedFunc* check_support = tvm::runtime::Registry::Get(support_func_name)) {
-      has_native_support = (*check_support)(compute_version);
+  if (target->kind->name == "cuda") {
+    if (const PackedFunc* get_cv =
+            tvm::runtime::Registry::Get("tvm.contrib.nvcc.get_compute_version")) {
+      std::string compute_version = (*get_cv)(target);
+      if (const PackedFunc* check_support = tvm::runtime::Registry::Get(support_func_name)) {
+        has_native_support = (*check_support)(compute_version);
+      }
     }
   }
   return has_native_support;