diff options
| -rw-r--r-- | xtask/src/main.rs | 2 | ||||
| -rw-r--r-- | zluda/src/impl/device.rs | 4 |
2 files changed, 5 insertions, 1 deletions
diff --git a/xtask/src/main.rs b/xtask/src/main.rs index d5e0cf8..3f1f224 100644 --- 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 index c7e8190..b7dd0f5 100644 --- 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(()); |