has_triton: Use the device interface for detecting Triton availability (#139171)

Summary:
This PR replaces the `has_triton()` global method which was previously used for this task.

X-link: pytorch/pytorch#139171
Approved by: https://github.com/jansel, https://github.com/shink

Reviewed By: huydhn

Differential Revision: D74338720

fbshipit-source-id: 27106df937bbdea2da1f4911ffffcfae056f844d

Author: generatedunixname499836121

userbenchmark/dynamo/dynamobench/_dynamo/utils.py

@@ -1546,7 +1546,7 @@ def record_compilation_metrics(
         "dynamo_config": _get_dynamo_config_for_logging(),
         "inductor_config": _scrubbed_inductor_config_for_logging(),
         "cuda_version": torch.version.cuda,
-        "triton_version": triton.__version__ if has_triton() else "",
+        "triton_version": triton.__version__ if has_triton_package() else "",
         "remote_cache_version": remote_cache_version,
         "inductor_fx_remote_cache_backend_type": inductor_fx_remote_cache_backend_type,
         "python_version": sys.version,
@@ -3803,15 +3803,14 @@ def build_checkpoint_variable(**options):
     )
 
 
-def is_compile_supported(device_type):
+def is_compile_supported(device_type: str) -> bool:
     from .eval_frame import is_dynamo_supported
 
-    type = torch.device(device_type).type
     compile_supported = is_dynamo_supported()
-    if type == "cpu":
+    if device_type == "cpu":
         pass
-    elif type in ["cuda", "xpu"] and compile_supported:
-        compile_supported = has_triton()
+    elif device_type in ["cuda", "xpu"] and compile_supported:
+        compile_supported = has_triton(device_type)
     else:
         compile_supported = False
     return compile_supported