From 90f0b054156a856f0dc0a1aa8763b7bb3c586195 Mon Sep 17 00:00:00 2001 From: tuna2134 Date: Wed, 11 Sep 2024 10:16:08 +0000 Subject: [PATCH] add directml support --- sbv2_api/Cargo.toml | 1 + 1 file changed, 1 insertion(+) diff --git a/sbv2_api/Cargo.toml b/sbv2_api/Cargo.toml index b1ac73a..11196c8 100644 --- a/sbv2_api/Cargo.toml +++ b/sbv2_api/Cargo.toml @@ -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"] \ No newline at end of file