fix: cuda serialization

r0.5
Meng Zhang 2023-11-07 00:55:38 -08:00
parent 0226a379de
commit 277528d01c
1 changed files with 1 additions and 0 deletions

View File

@ -76,6 +76,7 @@ pub enum Device {
Cpu,
#[cfg(feature = "cuda")]
#[strum(serialize = "cuda")]
Cuda,
#[cfg(all(target_os = "macos", target_arch = "aarch64"))]