Lint the code

Signed-off-by: Lu Fang <lufang@fb.com>

Author: houseroad

benchmarks/kernels/benchmark_moe.py

@@ -448,7 +448,8 @@ def tune(
             if visible_device != f"{self.device_id}":
                 need_device_guard = True
 
-        with torch.cuda.device(self.device_id) if need_device_guard else nullcontext():
+        with torch.cuda.device(
+                self.device_id) if need_device_guard else nullcontext():
             for config in tqdm(search_space):
                 try:
                     kernel_time = benchmark_config(
@@ -587,8 +588,7 @@ def main(args: argparse.Namespace):
         # Ray will set ROCR_VISIBLE_DEVICES for device visibility
         logger.warning(
             "Ray uses ROCR_VISIBLE_DEVICES to control device accessibility."
-            "Replacing HIP_VISIBLE_DEVICES with ROCR_VISIBLE_DEVICES."
-        )
+            "Replacing HIP_VISIBLE_DEVICES with ROCR_VISIBLE_DEVICES.")
         val = os.environ["HIP_VISIBLE_DEVICES"]
         os.environ["ROCR_VISIBLE_DEVICES"] = val
         del os.environ["HIP_VISIBLE_DEVICES"]

vllm/platforms/rocm.py

@@ -61,9 +61,9 @@
 _ROCM_DEVICE_ID_NAME_MAP: Dict[str, str] = {
     "0x74a0": "AMD_Instinct_MI300A",
     "0x74a1": "AMD_Instinct_MI300X",
-    "0x74b5": "AMD_Instinct_MI300X", # MI300X VF
+    "0x74b5": "AMD_Instinct_MI300X",  # MI300X VF
     "0x74a5": "AMD_Instinct_MI325X",
-    "0x74b9": "AMD_Instinct_MI325X", # MI325X VF
+    "0x74b9": "AMD_Instinct_MI325X",  # MI325X VF
     "0x74a9": "AMD_Instinct_MI300X_HF",
     "0x74bd": "AMD_Instinct_MI300X_HF",
 }