mirrors/ZLUDA
1
0
Fork 0
mirror of https://github.com/vosen/ZLUDA.git synced 2025-04-20 00:14:45 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Add clz

Browse source
This commit is contained in:
Andrzej Janik 2024-10-06 02:21:06 +02:00
parent 053c41fbb9
commit d173828492
1 changed files with 38 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

39
ptx/src/pass/emit_llvm.rs
View file

@ -486,7 +486,7 @@ impl<'a, 'input> MethodEmitContext<'a, 'input> {
ast::Instruction::Cos { data, arguments } => todo!(),
ast::Instruction::Lg2 { data, arguments } => todo!(),
ast::Instruction::Ex2 { data, arguments } => todo!(),
ast::Instruction::Clz { data, arguments } => todo!(),
ast::Instruction::Clz { data, arguments } => self.emit_clz(data, arguments),
ast::Instruction::Brev { data, arguments } => self.emit_brev(data, arguments),
ast::Instruction::Popc { data, arguments } => todo!(),
ast::Instruction::Xor { data, arguments } => todo!(),
@ -828,6 +828,43 @@ impl<'a, 'input> MethodEmitContext<'a, 'input> {
};
Ok(())
}
fn emit_clz(
&mut self,
data: ptx_parser::ScalarType,
arguments: ptx_parser::ClzArgs<SpirvWord>,
) -> Result<(), TranslateError> {
let llvm_fn = match data.size_of() {
4 => c"llvm.ctlz.i32",
8 => c"llvm.ctlz.i64",
_ => return Err(error_unreachable()),
};
let type_ = get_scalar_type(self.context, data.into());
let pred = get_scalar_type(self.context, ast::ScalarType::Pred);
let fn_type = get_function_type(
self.context,
iter::once(&ast::ScalarType::U32.into()),
[Ok(type_), Ok(pred)].into_iter(),
)?;
let mut fn_ = unsafe { LLVMGetNamedFunction(self.module, llvm_fn.as_ptr()) };
if fn_ == ptr::null_mut() {
fn_ = unsafe { LLVMAddFunction(self.module, llvm_fn.as_ptr(), fn_type) };
}
let src = self.resolver.value(arguments.src)?;
let false_ = unsafe { LLVMConstInt(pred, 0, 0) };
let mut args = [src, false_];
self.resolver.with_result(arguments.dst, |dst| unsafe {
LLVMBuildCall2(
self.builder,
fn_type,
fn_,
args.as_mut_ptr(),
args.len() as u32,
dst,
)
});
Ok(())
}
}
fn get_pointer_type<'ctx>(