mirrors/Odin
1
0
Fork 0
mirror of https://github.com/odin-lang/Odin.git synced 2026-08-18 19:32:15 +00:00
Code Issues Packages Projects Releases Wiki Activity

deinterleave -> shuffle

Browse Source
This commit is contained in:
kalsprite
2026-08-16 00:24:09 -07:00
parent 95c4004239
commit f7b734c82b
2 changed files with 55 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

25
src/llvm_backend_proc.cpp
View File

@@ -1601,7 +1601,6 @@ gb_internal lbValue lb_build_builtin_simd_proc(lbProcedure *p, Ast *expr, TypeAn
case BuiltinProc_simd_deinterleave:
{
lbValue arg0 = lb_build_expr(p, ce->args[0]);
LLVMTypeRef types[1] = {lb_type(m, arg0.type)};
GB_ASSERT(ce->args[1]->tav.value.kind == ExactValue_Integer);
int n = cast(int)exact_value_to_i64(ce->args[1]->tav.value);
@@ -1611,11 +1610,27 @@ gb_internal lbValue lb_build_builtin_simd_proc(lbProcedure *p, Ast *expr, TypeAn
return res;
}
gbString name = gb_string_make(heap_allocator(), "");
name = gb_string_append_fmt(name, "llvm.vector.deinterleave%d", n);
defer (gb_string_free(name));
// `llvm.vector.deinterleave<N>` for N > 2 cannot be selected or legalized on most
// targets, only arm64 takes it. The split is done with shuffles, same as
// `simd_interleave`. Output `j` is the input strided by N starting at lane `j`.
LLVMTypeRef llvm_u32 = lb_type(m, t_u32);
LLVMTypeRef vector_type = LLVMTypeOf(arg0.value);
LLVMValueRef undef = LLVMGetUndef(vector_type);
res.value = lb_call_intrinsic(p, name, &arg0.value, 1, types, gb_count_of(types));
unsigned width = LLVMGetVectorSize(vector_type);
unsigned part = width/cast(unsigned)n;
LLVMValueRef agg = LLVMGetUndef(lb_type(m, tv.type));
LLVMValueRef *mask = temporary_alloc_array<LLVMValueRef>(part);
for (int j = 0; j < n; j++) {
for (unsigned i = 0; i < part; i++) {
mask[i] = LLVMConstInt(llvm_u32, i*cast(unsigned)n + cast(unsigned)j, false);
}
LLVMValueRef lanes = LLVMBuildShuffleVector(p->builder, arg0.value, undef, LLVMConstVector(mask, part), "");
agg = LLVMBuildInsertValue(p->builder, agg, lanes, cast(unsigned)j, "");
}
res.value = agg;
return res;
}
}