[FRONTEND] fix the hard code builder.arch that could block third_part…

…y tests (triton-lang#1859) For CUDA devices, the `builder.arch` is an int. For third_party devices, this line would be a TypeError. For example: ``` TypeError: '<' not supported between instances of 'dict' and 'int' ``` Co-authored-by: Wang Weihan <[email protected]>
henryg-d-Matrix · Jul 12, 2023 · d50e32f · d50e32f
1 parent bf5acf4
commit d50e32f
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/python/triton/language/semantic.py b/python/triton/language/semantic.py
@@ -663,6 +663,11 @@ def bitcast(input: tl.tensor,
                      dst_ty)
 
 
+# TODO: architecture descriptor class
+def _is_cuda(arch):
+    return isinstance(arch, int)
+
+
 def cast(input: tl.tensor,
          dst_ty: tl.dtype,
          builder: ir.builder) -> tl.tensor:
@@ -677,7 +682,7 @@ def cast(input: tl.tensor,
     src_sca_ty = src_ty.scalar
     dst_sca_ty = dst_ty.scalar
 
-    if builder.arch < 89 and \
+    if _is_cuda(builder.arch) and builder.arch < 89 and \
        (src_sca_ty.is_fp8e4() or dst_sca_ty.is_fp8e4()):
         warnings.warn("Standard tl.float8e4 format will be deprecated on SM < 89. "
                       "Please use tl.float8e4b15.", DeprecationWarning)