ed/Odin
1
0
Fork 0
mirror of https://github.com/Ed94/Odin.git synced 2026-07-29 19:00:06 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add intrinsics.simd_masked_load and intrinsics.simd_masked_store

Browse Source
This commit is contained in:
gingerBill
2024-08-05 14:08:41 +01:00
parent 7e701d1677
commit 84ac56f778
6 changed files with 75 additions and 28 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/llvm_backend_proc.cpp
+15 -3
View File
@@ -1691,20 +1691,24 @@ gb_internal lbValue lb_build_builtin_simd_proc(lbProcedure *p, Ast *expr, TypeAn
case BuiltinProc_simd_gather:
case BuiltinProc_simd_scatter:
case BuiltinProc_simd_masked_load:
case BuiltinProc_simd_masked_store:
{
LLVMValueRef ptr = arg0.value;
LLVMValueRef val = arg1.value;
LLVMValueRef mask = arg2.value;
unsigned count = cast(unsigned)get_array_type_count(arg0.type);
unsigned count = cast(unsigned)get_array_type_count(arg1.type);
LLVMTypeRef mask_type = LLVMVectorType(LLVMInt1TypeInContext(p->module->ctx), count);
mask = LLVMBuildTrunc(p->builder, mask, mask_type, "");
char const *name = nullptr;
switch (builtin_id) {
case BuiltinProc_simd_gather: name = "llvm.masked.gather"; break;
case BuiltinProc_simd_scatter: name = "llvm.masked.scatter"; break;
case BuiltinProc_simd_gather: name = "llvm.masked.gather"; break;
case BuiltinProc_simd_scatter: name = "llvm.masked.scatter"; break;
case BuiltinProc_simd_masked_load: name = "llvm.masked.load"; break;
case BuiltinProc_simd_masked_store: name = "llvm.masked.store"; break;
}
LLVMTypeRef types[2] = {
lb_type(p->module, arg1.type),
@@ -1716,12 +1720,20 @@ gb_internal lbValue lb_build_builtin_simd_proc(lbProcedure *p, Ast *expr, TypeAn
LLVMValueRef args[4] = {};
switch (builtin_id) {
case BuiltinProc_simd_masked_load:
types[1] = lb_type(p->module, t_rawptr);
/*fallthrough*/
case BuiltinProc_simd_gather:
args[0] = ptr;
args[1] = align;
args[2] = mask;
args[3] = val;
// res.type = arg1.type;
break;
case BuiltinProc_simd_masked_store:
types[1] = lb_type(p->module, t_rawptr);
/*fallthrough*/
case BuiltinProc_simd_scatter:
args[0] = val;
args[1] = ptr;