spirv: make decl deps a hash map instead of an arraylist

The same declaration can be added to the dependency set multiple
times, and in this case we still need to emit it once. By making
this list a hash map instead, we can do that quite easily.

This commit also introduces some additional debug logging regarding
decls.
This commit is contained in:
Robin Voetter 2023-04-22 11:18:58 +02:00
parent d961b11cde
commit e26d8d0604
No known key found for this signature in database
GPG Key ID: E755662F227CB468
2 changed files with 15 additions and 12 deletions

View File

@ -238,6 +238,7 @@ pub const DeclGen = struct {
else => unreachable, else => unreachable,
}; };
const spv_decl_index = try self.resolveDecl(fn_decl_index); const spv_decl_index = try self.resolveDecl(fn_decl_index);
try self.func.decl_deps.put(self.spv.gpa, spv_decl_index, {});
return self.spv.declPtr(spv_decl_index).result_id; return self.spv.declPtr(spv_decl_index).result_id;
} }
@ -459,7 +460,7 @@ pub const DeclGen = struct {
/// If full, its flushed. /// If full, its flushed.
partial_word: std.BoundedArray(u8, @sizeOf(Word)) = .{}, partial_word: std.BoundedArray(u8, @sizeOf(Word)) = .{},
/// The declaration dependencies of the constant we are lowering. /// The declaration dependencies of the constant we are lowering.
decl_deps: std.ArrayList(SpvModule.Decl.Index), decl_deps: std.AutoArrayHashMap(SpvModule.Decl.Index, void),
/// Utility function to get the section that instructions should be lowered to. /// Utility function to get the section that instructions should be lowered to.
fn section(self: *@This()) *SpvSection { fn section(self: *@This()) *SpvSection {
@ -582,14 +583,15 @@ pub const DeclGen = struct {
// just generate an empty pointer. Function pointers are represented by usize for now, // just generate an empty pointer. Function pointers are represented by usize for now,
// though. // though.
try self.addInt(Type.usize, Value.initTag(.zero)); try self.addInt(Type.usize, Value.initTag(.zero));
// TODO: Add dependency
return; return;
}, },
.extern_fn => unreachable, // TODO .extern_fn => unreachable, // TODO
else => { else => {
const result_id = dg.spv.allocId(); const result_id = dg.spv.allocId();
log.debug("addDeclRef {s} = {}", .{ decl.name, result_id.id }); log.debug("addDeclRef: id = {}, index = {}, name = {s}", .{ result_id.id, @enumToInt(spv_decl_index), decl.name });
try self.decl_deps.append(spv_decl_index); try self.decl_deps.put(spv_decl_index, {});
const decl_id = dg.spv.declPtr(spv_decl_index).result_id; const decl_id = dg.spv.declPtr(spv_decl_index).result_id;
// TODO: Do we need a storage class cast here? // TODO: Do we need a storage class cast here?
@ -861,7 +863,8 @@ pub const DeclGen = struct {
assert(storage_class != .Generic and storage_class != .Function); assert(storage_class != .Generic and storage_class != .Function);
log.debug("lowerIndirectConstant: ty = {}, val = {}", .{ ty.fmt(self.module), val.fmtDebug() }); const var_id = self.spv.allocId();
log.debug("lowerIndirectConstant: id = {}, index = {}, ty = {}, val = {}", .{ var_id.id, @enumToInt(spv_decl_index), ty.fmt(self.module), val.fmtDebug() });
const section = &self.spv.globals.section; const section = &self.spv.globals.section;
@ -897,7 +900,7 @@ pub const DeclGen = struct {
.u32_ty_id = self.typeId(u32_ty_ref), .u32_ty_id = self.typeId(u32_ty_ref),
.members = std.ArrayList(SpvType.Payload.Struct.Member).init(self.gpa), .members = std.ArrayList(SpvType.Payload.Struct.Member).init(self.gpa),
.initializers = std.ArrayList(IdRef).init(self.gpa), .initializers = std.ArrayList(IdRef).init(self.gpa),
.decl_deps = std.ArrayList(SpvModule.Decl.Index).init(self.gpa), .decl_deps = std.AutoArrayHashMap(SpvModule.Decl.Index, void).init(self.gpa),
}; };
defer icl.members.deinit(); defer icl.members.deinit();
@ -917,7 +920,6 @@ pub const DeclGen = struct {
.constituents = icl.initializers.items, .constituents = icl.initializers.items,
}); });
const var_id = self.spv.allocId();
self.spv.globalPtr(spv_decl_index).?.result_id = var_id; self.spv.globalPtr(spv_decl_index).?.result_id = var_id;
try section.emit(self.spv.gpa, .OpVariable, .{ try section.emit(self.spv.gpa, .OpVariable, .{
.id_result_type = self.typeId(ptr_constant_struct_ty_ref), .id_result_type = self.typeId(ptr_constant_struct_ty_ref),
@ -951,7 +953,7 @@ pub const DeclGen = struct {
}); });
} }
try self.spv.declareDeclDeps(spv_decl_index, icl.decl_deps.items); try self.spv.declareDeclDeps(spv_decl_index, icl.decl_deps.keys());
self.spv.endGlobal(spv_decl_index, begin_inst); self.spv.endGlobal(spv_decl_index, begin_inst);
} }
@ -1007,7 +1009,8 @@ pub const DeclGen = struct {
false, false,
alignment, alignment,
); );
try self.func.decl_deps.append(self.spv.gpa, spv_decl_index); log.debug("indirect constant: index = {}", .{@enumToInt(spv_decl_index)});
try self.func.decl_deps.put(self.spv.gpa, spv_decl_index, {});
try self.func.body.emit(self.spv.gpa, .OpLoad, .{ try self.func.body.emit(self.spv.gpa, .OpLoad, .{
.id_result_type = result_ty_id, .id_result_type = result_ty_id,
@ -1471,7 +1474,7 @@ pub const DeclGen = struct {
const spv_decl_index = try self.resolveDecl(self.decl_index); const spv_decl_index = try self.resolveDecl(self.decl_index);
const decl_id = self.spv.declPtr(spv_decl_index).result_id; const decl_id = self.spv.declPtr(spv_decl_index).result_id;
log.debug("genDecl {s} = {}", .{ decl.name, decl_id }); log.debug("genDecl: id = {}, index = {}, name = {s}", .{ decl_id.id, @enumToInt(spv_decl_index), decl.name });
if (decl.val.castTag(.function)) |_| { if (decl.val.castTag(.function)) |_| {
assert(decl.ty.zigTypeTag() == .Fn); assert(decl.ty.zigTypeTag() == .Fn);

View File

@ -40,14 +40,14 @@ pub const Fn = struct {
/// the end of this function definition. /// the end of this function definition.
body: Section = .{}, body: Section = .{},
/// The decl dependencies that this function depends on. /// The decl dependencies that this function depends on.
decl_deps: std.ArrayListUnmanaged(Decl.Index) = .{}, decl_deps: std.AutoArrayHashMapUnmanaged(Decl.Index, void) = .{},
/// Reset this function without deallocating resources, so that /// Reset this function without deallocating resources, so that
/// it may be used to emit code for another function. /// it may be used to emit code for another function.
pub fn reset(self: *Fn) void { pub fn reset(self: *Fn) void {
self.prologue.reset(); self.prologue.reset();
self.body.reset(); self.body.reset();
self.decl_deps.items.len = 0; self.decl_deps.clearRetainingCapacity();
} }
/// Free the resources owned by this function. /// Free the resources owned by this function.
@ -358,7 +358,7 @@ pub fn flush(self: *Module, file: std.fs.File) !void {
pub fn addFunction(self: *Module, decl_index: Decl.Index, func: Fn) !void { pub fn addFunction(self: *Module, decl_index: Decl.Index, func: Fn) !void {
try self.sections.functions.append(self.gpa, func.prologue); try self.sections.functions.append(self.gpa, func.prologue);
try self.sections.functions.append(self.gpa, func.body); try self.sections.functions.append(self.gpa, func.body);
try self.declareDeclDeps(decl_index, func.decl_deps.items); try self.declareDeclDeps(decl_index, func.decl_deps.keys());
} }
/// Fetch the result-id of an OpString instruction that encodes the path of the source /// Fetch the result-id of an OpString instruction that encodes the path of the source