mirror/zig
1
0
Fork 0
mirror of https://github.com/ziglang/zig.git synced 2026-02-01 04:03:40 +00:00
Code Issues Packages Projects Releases Wiki Activity

allow global/local/shared address spaces on amdgcn

Browse Source
This commit is contained in:
Robin Voetter 2022-08-20 12:42:27 +02:00
parent 5859d8458f
commit 3392de87dc
No known key found for this signature in database
GPG Key ID: E755662F227CB468
2 changed files with 12 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
src/Sema.zig
View File

@ -30303,13 +30303,15 @@ pub fn analyzeAddrspace(
const address_space = addrspace_tv.val.toEnum(std.builtin.AddressSpace);
const target = sema.mod.getTarget();
const arch = target.cpu.arch;
const is_gpu = arch == .nvptx or arch == .nvptx64;
const is_nv = arch == .nvptx or arch == .nvptx64;
const is_gpu = is_nv or arch == .amdgcn;
const supported = switch (address_space) {
.generic => true,
.gs, .fs, .ss => (arch == .i386 or arch == .x86_64) and ctx == .pointer,
// TODO: check that .shared and .local are left uninitialized
.global, .param, .shared, .local => is_gpu,
.param => is_nv,
.global, .shared, .local => is_gpu,
.constant => is_gpu and (ctx == .constant),
};

8
src/codegen/llvm.zig
View File

@ -2659,6 +2659,14 @@ pub const DeclGen = struct {
.local => llvm.address_space.nvptx.local,
else => unreachable,
},
.amdgcn => switch (address_space) {
.generic => llvm.address_space.flat,
.global => llvm.address_space.amdgpu.global,
.constant => llvm.address_space.amdgpu.constant,
.shared => llvm.address_space.amdgpu.local,
.local => llvm.address_space.amdgpu.private,
else => unreachable,
}.
else => switch (address_space) {
.generic => llvm.address_space.default,
else => unreachable,