fix: cuda serialization

refactor-extract-code
Meng Zhang 2023-11-07 00:55:38 -08:00 committed by GitHub
parent cd8edc2fef
commit 3c3b14c9f5
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
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"))]