mirror/zig
1
0
Fork 0
mirror of https://github.com/ziglang/zig.git synced 2025-12-06 06:13:07 +00:00
Code Issues Packages Projects Releases Wiki Activity

spirv: air splat

Browse Source
This commit is contained in:
Robin Voetter 2024-01-21 22:48:31 +01:00
parent 76d5696434
commit 1d548aa2aa
No known key found for this signature in database
2 changed files with 15 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
src/codegen/spirv.zig
View File

@ -2195,6 +2195,7 @@ const DeclGen = struct {
.mul_add => try self.airMulAdd(inst), .mul_add => try self.airMulAdd(inst),
.splat => try self.airSplat(inst),
.reduce, .reduce_optimized => try self.airReduce(inst), .reduce, .reduce_optimized => try self.airReduce(inst),
.shuffle => try self.airShuffle(inst), .shuffle => try self.airShuffle(inst),
@ -2603,6 +2604,7 @@ const DeclGen = struct {
// Idk why spir-v doesn't have a dedicated abs() instruction in the base // Idk why spir-v doesn't have a dedicated abs() instruction in the base
// instruction set. For now we're just going to negate and check to avoid // instruction set. For now we're just going to negate and check to avoid
// importing the extinst. // importing the extinst.
// TODO: Make this a call to compiler rt / ext inst
const neg_id = self.spv.allocId(); const neg_id = self.spv.allocId();
const args = .{ const args = .{
.id_result_type = self.typeId(operand_scalar_ty_ref), .id_result_type = self.typeId(operand_scalar_ty_ref),
@ -2877,6 +2879,19 @@ const DeclGen = struct {
return try wip.finalize(); return try wip.finalize();
} }
fn airSplat(self: *DeclGen, inst: Air.Inst.Index) !?IdRef {
if (self.liveness.isUnused(inst)) return null;
const ty_op = self.air.instructions.items(.data)[@intFromEnum(inst)].ty_op;
const operand_id = try self.resolve(ty_op.operand);
const result_ty = self.typeOfIndex(inst);
var wip = try self.elementWise(result_ty);
defer wip.deinit();
for (wip.results) |*result_id| {
result_id.* = operand_id;
}
return try wip.finalize();
}
fn airReduce(self: *DeclGen, inst: Air.Inst.Index) !?IdRef { fn airReduce(self: *DeclGen, inst: Air.Inst.Index) !?IdRef {
if (self.liveness.isUnused(inst)) return null; if (self.liveness.isUnused(inst)) return null;
const mod = self.module; const mod = self.module;

1
test/behavior/vector.zig
View File

@ -326,7 +326,6 @@ test "vector @splat" {
if (builtin.zig_backend == .stage2_aarch64) return error.SkipZigTest; // TODO if (builtin.zig_backend == .stage2_aarch64) return error.SkipZigTest; // TODO
if (builtin.zig_backend == .stage2_arm) return error.SkipZigTest; // TODO if (builtin.zig_backend == .stage2_arm) return error.SkipZigTest; // TODO
if (builtin.zig_backend == .stage2_sparc64) return error.SkipZigTest; // TODO if (builtin.zig_backend == .stage2_sparc64) return error.SkipZigTest; // TODO
if (builtin.zig_backend == .stage2_spirv64) return error.SkipZigTest;
if (builtin.zig_backend == .stage2_llvm and if (builtin.zig_backend == .stage2_llvm and
builtin.os.tag == .macos) builtin.os.tag == .macos)