Skip to content

[BUG] Runtime default_ptx_arch setting needs to consider NVRTC supported architectures #858

@shi-eric

Description

@shi-eric

Bug Description

Currently, it is possible to ask NVRTC to compile code for an architecture it does not recognize, when it should really be falling back to the default minimum architecture value of 75.

This appears to be due to the following code that sets the runtime.default_ptx_arch to the lowest value of architectures on the system without checking that each architecture is supported by NVRTC.

warp/warp/context.py

Lines 4098 to 4100 in 3889aec

eligible_archs = [d.arch for d in self.cuda_devices if d.arch >= self.default_ptx_arch]
if eligible_archs:
self.default_ptx_arch = min(eligible_archs)

System Information

No response

Metadata

Metadata

Assignees

Labels

bugSomething isn't working

Type

Projects

No projects

Milestone

Relationships

None yet

Development

No branches or pull requests

Issue actions