fix: use correct DirectML device name

pykeio · Dec 23, 2024 · 46c3376 · 46c3376
1 parent ec24307
commit 46c3376
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/memory.rs b/src/memory.rs
@@ -253,7 +253,7 @@ impl AllocationDevice {
 	pub const CUDA_PINNED: AllocationDevice = AllocationDevice("CudaPinned");
 	pub const CANN: AllocationDevice = AllocationDevice("Cann");
 	pub const CANN_PINNED: AllocationDevice = AllocationDevice("CannPinned");
-	pub const DIRECTML: AllocationDevice = AllocationDevice("Dml");
+	pub const DIRECTML: AllocationDevice = AllocationDevice("DML");
 	pub const DIRECTML_CPU: AllocationDevice = AllocationDevice("DML CPU");
 	pub const HIP: AllocationDevice = AllocationDevice("Hip");
 	pub const HIP_PINNED: AllocationDevice = AllocationDevice("HipPinned");