tocmo0nlord/axolotl
1
0
Fork 0
Code Issues Pull Requests Actions 3 Packages Projects Releases Wiki Activity

remove duplicate info

Browse Source
This commit is contained in:
Dan Saunders
2025-02-24 20:02:16 +00:00
parent ffd2985750
commit e9df4444be
1 changed files with 0 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
src/axolotl/telemetry/manager.py
View File

@@ -248,7 +248,6 @@ class TelemetryManager:
{ {
"name": torch.cuda.get_device_name(i), "name": torch.cuda.get_device_name(i),
"memory": torch.cuda.get_device_properties(i).total_memory, "memory": torch.cuda.get_device_properties(i).total_memory,
"type": "cuda",
} }
) )
@@ -262,7 +261,6 @@ class TelemetryManager:
"memory": torch.hip.get_device_properties(i).total_memory "memory": torch.hip.get_device_properties(i).total_memory
if hasattr(torch.hip, "get_device_properties") if hasattr(torch.hip, "get_device_properties")
else None, else None,
"type": "hip",
} }
) )
@@ -274,7 +272,6 @@ class TelemetryManager:
"name": "Apple Silicon", "name": "Apple Silicon",
# NOTE: this is memory allocated to this process, not total memory # NOTE: this is memory allocated to this process, not total memory
"memory": torch.mps.driver_allocated_memory(), "memory": torch.mps.driver_allocated_memory(),
"type": "mps",
} }
) )
@@ -290,7 +287,6 @@ class TelemetryManager:
{ {
"name": torch.xpu.get_device_name(i), "name": torch.xpu.get_device_name(i),
"memory": memory, "memory": memory,
"type": "xpu",
} }
) )
@@ -306,7 +302,6 @@ class TelemetryManager:
{ {
"name": torch.npu.get_device_name(i), "name": torch.npu.get_device_name(i),
"memory": memory, "memory": memory,
"type": "npu",
} }
) )