Skip to content

Commit a9bf36b

Browse files
jhuber6rorth
authored andcommitted
[Offload] Optimistically accept SM architectures (llvm#142399)
Summary: We try to clamp these to ones known to work, but we should probably just optimistically accept these. I'd prefer to update the flag check, but since NVIDIA refuses to publish their ELF format it's too much effort to reverse engineer. Fixes: llvm#138532
1 parent ee02b9d commit a9bf36b

File tree

1 file changed

+1
-2
lines changed
  • offload/plugins-nextgen/common/src/Utils

1 file changed

+1
-2
lines changed

offload/plugins-nextgen/common/src/Utils/ELF.cpp

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -75,8 +75,7 @@ checkMachineImpl(const object::ELFObjectFile<ELFT> &ELFObj, uint16_t EMachine) {
7575
} else if (Header.e_machine == EM_CUDA) {
7676
if (~Header.e_flags & EF_CUDA_64BIT_ADDRESS)
7777
return createError("Invalid CUDA addressing mode");
78-
if ((Header.e_flags & EF_CUDA_SM) < EF_CUDA_SM35 ||
79-
(Header.e_flags & EF_CUDA_SM) > EF_CUDA_SM90)
78+
if ((Header.e_flags & EF_CUDA_SM) < EF_CUDA_SM35)
8079
return createError("Unsupported NVPTX architecture");
8180
}
8281

0 commit comments

Comments
 (0)