Merge pull request #192 from SludgePhD/wgpu-17

webonnx · Oct 17, 2023 · c65c3e0 · c65c3e0
2 parents 564efaa + 3e9b0f5
commit c65c3e0
Show file tree

Hide file tree

Showing 5 changed files with 59 additions and 32 deletions.
diff --git a/Cargo.lock b/Cargo.lock
diff --git a/wonnx-cli/Cargo.toml b/wonnx-cli/Cargo.toml
@@ -30,7 +30,7 @@ protobuf = { version = "2.27.1", features = ["with-bytes"] }
 structopt = { version = "0.3.26", features = ["paw"] }
 thiserror = "1.0.31"
 tract-onnx = { version = "0.19.12", optional = true }
-wgpu = "0.16.0"
+wgpu = "0.17.1"
 wonnx = { version = "^0.5.1" }
 wonnx-preprocessing = { version = "^0.5.1" }
 human_bytes = "0.4.1"

diff --git a/wonnx-preprocessing/Cargo.toml b/wonnx-preprocessing/Cargo.toml
@@ -20,7 +20,7 @@ protobuf = { version = "2.27.1", features = ["with-bytes"] }
 thiserror = "1.0.31"
 tokenizers = "0.13.3"
 tract-onnx = { version = "0.19.2", optional = true }
-wgpu = "0.16.0"
+wgpu = "0.17.1"
 wonnx = { version = "^0.5.1" }
 serde_json = "^1.0"
 bytemuck = "1.9.1"

diff --git a/wonnx/Cargo.toml b/wonnx/Cargo.toml
@@ -22,7 +22,7 @@ exclude = [
 ]
 
 [dependencies]
-wgpu = "0.16.0"
+wgpu = "0.17.1"
 bytemuck = { version = "1.9.1", features = ["extern_crate_alloc"] }
 protobuf = { version = "2.27.1", features = ["with-bytes"] }
 log = "0.4.17"

diff --git a/wonnx/src/resource.rs b/wonnx/src/resource.rs
@@ -8,7 +8,7 @@ pub async fn request_device_queue() -> (wgpu::Device, wgpu::Queue) {
         dx12_shader_compiler: wgpu::Dx12Compiler::Fxc,
     };
     let instance = wgpu::Instance::new(instance_descriptor);
-    let adapter = wgpu::util::initialize_adapter_from_env_or_default(&instance, backends, None)
+    let adapter = wgpu::util::initialize_adapter_from_env_or_default(&instance, None)
         .await
         .expect("No GPU found given preference");