Correctly report emulated wave32 CUDA device (#216)

vosen · Apr 29, 2024 · bdc652f · bdc652f
1 parent 995bc95
commit bdc652f
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 1 deletion.
diff --git a/xtask/src/main.rs b/xtask/src/main.rs
@@ -322,7 +322,7 @@ mod os {
         }
     }
 
-    pub fn zip(workspace: Workspace) -> i32 {
+    pub(crate) fn zip(workspace: Workspace) -> i32 {
         let mut target_file = workspace.target_directory.clone();
         target_file.push("zluda.tar.gz");
         let gz_file = File::create(target_file).unwrap();

diff --git a/zluda/src/impl/device.rs b/zluda/src/impl/device.rs
@@ -109,6 +109,10 @@ pub(crate) unsafe fn get_attribute(
         return Err(CUresult::CUDA_ERROR_INVALID_VALUE);
     }
     let hip_attrib = match attrib {
+        CUdevice_attribute::CU_DEVICE_ATTRIBUTE_WARP_SIZE => {
+            *pi = 32;
+            return Ok(());
+        }
         CUdevice_attribute::CU_DEVICE_ATTRIBUTE_ASYNC_ENGINE_COUNT => {
             *pi = 1;
             return Ok(());