fix rocm api memory measurment

huggingface · Feb 14, 2024 · 019cbad · 019cbad
1 parent ef50b83
commit 019cbad
Showing 1 changed file with 0 additions and 2 deletions.
diff --git a/tests/test_api.py b/tests/test_api.py
@@ -97,8 +97,6 @@ def test_api_memory_tracker(device, backend):
         measured_memory = final_memory.max_allocated - initial_memory.max_allocated
     elif device == "cuda":
         measured_memory = final_memory.max_vram - initial_memory.max_vram
-        if torch.version.hip is not None:
-            measured_memory -= 1600  # something is wrong with amdsmi or rocm
     else:
         measured_memory = final_memory.max_ram - initial_memory.max_ram