add directml support

This commit is contained in:
tuna2134
2024-09-11 10:16:08 +00:00
parent a92dd07314
commit 90f0b05415

View File

@@ -17,3 +17,4 @@ tokio = { version = "1.40.0", features = ["full"] }
cuda = ["sbv2_core/cuda"]
cuda_tf32 = ["sbv2_core/cuda_tf32"]
dynamic = ["sbv2_core/dynamic"]
directml = ["sbv2_core/directml"]