Skip to content

Commit eb9ed93

Browse files
authored
[Offload] Optimistically accept SM architectures (#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: #138532
1 parent c005df3 commit eb9ed93

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)