Applied-Software/ZLUDA
5
0
Fork 0
mirror of https://github.com/vosen/ZLUDA.git synced 2025-07-17 09:16:23 +03:00
Code Issues Packages Projects Releases Wiki Activity

Fix handling of kernel args in stateful conversion

Browse Source
This commit is contained in:
Andrzej Janik
2021-06-11 12:36:23 +02:00
parent f0771e1fb6
commit 2198862e76
1 changed files with 8 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
ptx/src/translate.rs
View File

@ -4136,10 +4136,13 @@ fn emit_implicit_conversion(
let dst_type = map.get_or_add(builder, SpirvType::new(cv.to_type.clone())); let dst_type = map.get_or_add(builder, SpirvType::new(cv.to_type.clone()));
builder.convert_ptr_to_u(dst_type, Some(cv.dst), cv.src)?; builder.convert_ptr_to_u(dst_type, Some(cv.dst), cv.src)?;
} }
(TypeKind::Pointer, TypeKind::Scalar, &ConversionKind::Default) (TypeKind::Pointer, TypeKind::Scalar, &ConversionKind::Default) => {
| (TypeKind::Scalar, TypeKind::Pointer, &ConversionKind::Default) => { let result_type = map.get_or_add(builder, SpirvType::new(cv.to_type.clone()));
let dst_type = map.get_or_add(builder, SpirvType::new(cv.to_type.clone())); builder.convert_ptr_to_u(result_type, Some(cv.dst), cv.src)?;
builder.bitcast(dst_type, Some(cv.dst), cv.src)?; }
(TypeKind::Scalar, TypeKind::Pointer, &ConversionKind::Default) => {
let result_type = map.get_or_add(builder, SpirvType::new(cv.to_type.clone()));
builder.convert_u_to_ptr(result_type, Some(cv.dst), cv.src)?;
} }
_ => unreachable!(), _ => unreachable!(),
} }
@ -4478,7 +4481,7 @@ fn convert_to_stateful_memory_access<'a, 'input>(
for arg in (*method_decl).input_arguments.iter_mut() { for arg in (*method_decl).input_arguments.iter_mut() {
let new_id = id_defs.register_variable( let new_id = id_defs.register_variable(
ast::Type::Pointer(ast::ScalarType::U8, ast::StateSpace::Global), ast::Type::Pointer(ast::ScalarType::U8, ast::StateSpace::Global),
ast::StateSpace::Reg, ast::StateSpace::Param,
); );
let old_name = arg.name; let old_name = arg.name;
if func_args_ptr.contains(&arg.name) { if func_args_ptr.contains(&arg.name) {