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

Do the same for max

Browse source
This commit is contained in:
Violet 2025-09-12 20:51:45 +00:00
commit 85e8e61f2f
1 changed files with 38 additions and 14 deletions
Download patch file Download diff file Expand all files Collapse all files

52
ptx/src/pass/llvm/emit.rs
View file

@ -2305,11 +2305,9 @@ impl<'a> MethodEmitContext<'a> {
)
});
} else {
self.resolver.with_result(arguments.dst, |dst| unsafe {
let dst = CStr::from_ptr(dst);
LLVMSetValueName2(min, dst.as_ptr(), dst.count_bytes());
min
});
let name = self.resolver.get_or_add(arguments.dst);
unsafe { LLVMSetValueName2(min, name.as_ptr().cast(), name.len()) };
self.resolver.register(arguments.dst, min);
}
Ok(())
}
@ -2322,22 +2320,48 @@ impl<'a> MethodEmitContext<'a> {
let llvm_prefix = match data {
ptx_parser::MinMaxDetails::Signed(..) => "llvm.smax",
ptx_parser::MinMaxDetails::Unsigned(..) => "llvm.umax",
ptx_parser::MinMaxDetails::Float(ptx_parser::MinMaxFloat { nan: true, .. }) => {
"llvm.maximum"
}
ptx_parser::MinMaxDetails::Float(ptx_parser::MinMaxFloat { .. }) => "llvm.maxnum",
};
let intrinsic = format!("{}.{}\0", llvm_prefix, LLVMTypeDisplay(data.type_()));
let llvm_type = get_scalar_type(self.context, data.type_());
self.emit_intrinsic(
let a = self.resolver.value(arguments.src1)?;
let b = self.resolver.value(arguments.src2)?;
let min = self.emit_intrinsic(
unsafe { CStr::from_bytes_with_nul_unchecked(intrinsic.as_bytes()) },
Some(arguments.dst),
None,
Some(&data.type_().into()),
vec![
(self.resolver.value(arguments.src1)?, llvm_type),
(self.resolver.value(arguments.src2)?, llvm_type),
],
vec![(a, llvm_type), (b, llvm_type)],
)?;
if let ptx_parser::MinMaxDetails::Float(ptx_parser::MinMaxFloat {
nan: true, type_, ..
}) = data
{
let is_nan = unsafe {
LLVMBuildFCmp(
self.builder,
LLVMRealPredicate::LLVMRealUNO,
a,
b,
LLVM_UNNAMED.as_ptr(),
)
};
self.resolver.with_result(arguments.dst, |dst| unsafe {
LLVMBuildSelect(
self.builder,
is_nan,
LLVMConstReal(get_scalar_type(self.context, type_), f64::NAN),
min,
dst,
)
});
} else {
let name = self.resolver.get_or_add(arguments.dst);
unsafe { LLVMSetValueName2(min, name.as_ptr().cast(), name.len()) };
self.resolver.register(arguments.dst, min);
}
Ok(())
}