Fix is_compile_supported() when device_type contains device index (#147837)

Summary:
Fixes #147826

X-link: pytorch/pytorch#147837
Approved by: https://github.com/anijain2305

Reviewed By: yangw-dev

Differential Revision: D72080412

fbshipit-source-id: ce80f0c151f1b96d58da03377a68a695037407bb

Author: generatedunixname499836121

userbenchmark/dynamo/dynamobench/_dynamo/utils.py

@@ -3751,10 +3751,11 @@ def build_checkpoint_variable(**options):
 def is_compile_supported(device_type):
     from .eval_frame import is_dynamo_supported
 
+    type = torch.device(device_type).type
     compile_supported = is_dynamo_supported()
-    if device_type == "cpu":
+    if type == "cpu":
         pass
-    elif device_type in ["cuda", "xpu"] and compile_supported:
+    elif type in ["cuda", "xpu"] and compile_supported:
         compile_supported = has_triton()
     else:
         compile_supported = False