From 5cce1a2edb58de0cc0fdc3b6e926d837131cccce Mon Sep 17 00:00:00 2001
From: Rukai <rubickent@gmail.com>
Date: Wed, 8 May 2019 21:50:44 +1000
Subject: [PATCH] Add multithreaded_compute test

---
 wgpu-rs/tests/multithreaded_compute.rs | 100 +++++++++++++++++++++++++
 1 file changed, 100 insertions(+)
 create mode 100644 wgpu-rs/tests/multithreaded_compute.rs

diff --git a/wgpu-rs/tests/multithreaded_compute.rs b/wgpu-rs/tests/multithreaded_compute.rs
new file mode 100644
index 0000000000..76ced61595
--- /dev/null
+++ b/wgpu-rs/tests/multithreaded_compute.rs
@@ -0,0 +1,100 @@
+#[test]
+#[cfg(any(feature = "vulkan", feature = "metal", feature = "dx12"))]
+fn multithreaded_compute() {
+    use std::thread;
+    use std::time::Duration;
+    use std::sync::mpsc;
+
+    let thread_count = 8;
+
+    let (tx, rx) = mpsc::channel();
+    for _ in 0..thread_count {
+        let tx = tx.clone();
+        thread::spawn(move || {
+            let numbers = vec!(100, 100, 100);
+
+            let size = (numbers.len() * std::mem::size_of::<u32>()) as u32;
+
+            let instance = wgpu::Instance::new();
+            let adapter = instance.get_adapter(&wgpu::AdapterDescriptor {
+                power_preference: wgpu::PowerPreference::Default,
+            });
+            let mut device = adapter.create_device(&wgpu::DeviceDescriptor {
+                extensions: wgpu::Extensions {
+                    anisotropic_filtering: false,
+                },
+            });
+
+            let cs_bytes = include_bytes!("./../../examples/data/collatz.comp.spv");
+            let cs_module = device.create_shader_module(cs_bytes);
+
+            let staging_buffer = device
+                .create_buffer_mapped(
+                    numbers.len(),
+                    wgpu::BufferUsageFlags::MAP_READ
+                        | wgpu::BufferUsageFlags::TRANSFER_DST
+                        | wgpu::BufferUsageFlags::TRANSFER_SRC,
+                )
+                .fill_from_slice(&numbers);
+
+            let storage_buffer = device.create_buffer(&wgpu::BufferDescriptor {
+                size,
+                usage: wgpu::BufferUsageFlags::STORAGE
+                    | wgpu::BufferUsageFlags::TRANSFER_DST
+                    | wgpu::BufferUsageFlags::TRANSFER_SRC,
+            });
+
+            let bind_group_layout = device.create_bind_group_layout(&wgpu::BindGroupLayoutDescriptor {
+                bindings: &[wgpu::BindGroupLayoutBinding {
+                    binding: 0,
+                    visibility: wgpu::ShaderStageFlags::COMPUTE,
+                    ty: wgpu::BindingType::StorageBuffer,
+                }],
+            });
+
+            let bind_group = device.create_bind_group(&wgpu::BindGroupDescriptor {
+                layout: &bind_group_layout,
+                bindings: &[wgpu::Binding {
+                    binding: 0,
+                    resource: wgpu::BindingResource::Buffer {
+                        buffer: &storage_buffer,
+                        range: 0..size,
+                    },
+                }],
+            });
+
+            let pipeline_layout = device.create_pipeline_layout(&wgpu::PipelineLayoutDescriptor {
+                bind_group_layouts: &[&bind_group_layout],
+            });
+
+            let compute_pipeline = device.create_compute_pipeline(&wgpu::ComputePipelineDescriptor {
+                layout: &pipeline_layout,
+                compute_stage: wgpu::PipelineStageDescriptor {
+                    module: &cs_module,
+                    entry_point: "main",
+                },
+            });
+
+            let mut encoder = device.create_command_encoder(&wgpu::CommandEncoderDescriptor { todo: 0 });
+            encoder.copy_buffer_to_buffer(&staging_buffer, 0, &storage_buffer, 0, size);
+            {
+                let mut cpass = encoder.begin_compute_pass();
+                cpass.set_pipeline(&compute_pipeline);
+                cpass.set_bind_group(0, &bind_group, &[]);
+                cpass.dispatch(numbers.len() as u32, 1, 1);
+            }
+            encoder.copy_buffer_to_buffer(&storage_buffer, 0, &staging_buffer, 0, size);
+
+            device.get_queue().submit(&[encoder.finish()]);
+
+            staging_buffer.map_read_async(0, size, |result: wgpu::BufferMapAsyncResult<&[u32]>| {
+                assert_eq!(result.unwrap().data, [25, 25, 25]);
+            });
+            tx.send(true).unwrap();
+        });
+    }
+
+    for _ in 0..thread_count {
+        rx.recv_timeout(Duration::from_secs(10)).expect("A thread never completed.");
+    }
+}