update workflows

.github/workflows/test_api_cuda.yaml

@@ -22,7 +22,7 @@ jobs:
             { torch_cuda: cu118, cuda_version: 11.8.0 },
           ]
 
-    runs-on: hf-dgx-01
+    runs-on: nvidia-gpu
     steps:
       - name: Checkout
         uses: actions/checkout@v3

.github/workflows/test_api_rocm.yaml

@@ -22,7 +22,7 @@ jobs:
             { torch_rocm: rocm5.7, torch_pre_release: 1, rocm_version: 5.7 },
           ]
 
-    runs-on: hf-amd-mi210-dev
+    runs-on: amd-gpu
     steps:
       - name: Checkout code
         uses: actions/checkout@v3

.github/workflows/test_cli_cuda_onnxruntime.yaml

@@ -13,7 +13,7 @@ concurrency:
 
 jobs:
   build_image_and_run_cli_cuda_onnxruntime_tests:
-    runs-on: hf-dgx-01
+    runs-on: nvidia-gpu
     steps:
       - name: Checkout
         uses: actions/checkout@v3

.github/workflows/test_cli_cuda_pytorch.yaml

@@ -22,7 +22,7 @@ jobs:
             { torch_cuda: cu118, cuda_version: 11.8.0 },
           ]
 
-    runs-on: hf-dgx-01
+    runs-on: nvidia-gpu
     steps:
       - name: Checkout
         uses: actions/checkout@v3

.github/workflows/test_cli_cuda_torch_ort.yaml

@@ -13,7 +13,7 @@ concurrency:
 
 jobs:
   build_image_and_run_cli_cuda_torch_ort_tests:
-    runs-on: hf-dgx-01
+    runs-on: nvidia-gpu
     steps:
       - name: Checkout
         uses: actions/checkout@v3

.github/workflows/test_cli_rocm_onnxruntime.yaml

@@ -13,7 +13,7 @@ concurrency:
 
 jobs:
   build_image_and_run_cli_rocm_onnxruntime_tests:
-    runs-on: hf-amd-mi210-dev
+    runs-on: amd-gpu
     steps:
       - name: Checkout
         uses: actions/checkout@v3

.github/workflows/test_cli_rocm_pytorch.yaml

@@ -22,7 +22,7 @@ jobs:
             { torch_rocm: rocm5.7, torch_pre_release: 1, rocm_version: 5.7 },
           ]
 
-    runs-on: hf-amd-mi210-dev
+    runs-on: [amd-gpu]
     steps:
       - name: Checkout code
         uses: actions/checkout@v3

.github/workflows/test_cli_tensorrt_llm.yaml

@@ -13,7 +13,7 @@ concurrency:
 
 jobs:
   pull_image_and_run_cli_tensorrt_llm_tests:
-    runs-on: hf-dgx-01
+    runs-on: nvidia-gpu
     steps:
       - name: Checkout
         uses: actions/checkout@v3

.github/workflows/test_cli_tensorrt_onnxruntime.yaml

@@ -13,7 +13,7 @@ concurrency:
 
 jobs:
   build_image_and_run_cli_tensorrt_onnxruntime_tests:
-    runs-on: hf-dgx-01
+    runs-on: nvidia-gpu
     steps:
       - name: Checkout
         uses: actions/checkout@v3

optimum_benchmark/trackers/latency.py

@@ -115,10 +115,7 @@ def __init__(self, device: str, backend: str):
         self.device = device
         self.backend = backend
 
-        if is_torch_distributed_available() and torch.distributed.is_initialized():
-            self.distributed = True
-        else:
-            self.distributed = False
+        self.distributed = is_torch_distributed_available() and torch.distributed.is_initialized()
 
         self.start_events: List[Union[float, torch.cuda.Event]] = []
         self.end_events: List[Union[float, torch.cuda.Event]] = []