Inline MLP example: Fix issue with CUDA arch versions on Docker containers

shader-slang / slang-python

Superseded by github.com/shader-slang/slang-torch

MIT License

26 stars 3 forks source link

Inline MLP example: Fix issue with CUDA arch versions on Docker containers #15

Closed saipraveenb25 closed 7 months ago

saipraveenb25 commented 7 months ago

Removes unsupported versions from TORCH_CUDA_ARCH_LIST environment variable, if present.

Ideally, we would simply specify the supported versions as extra_cflags when creating the torch extension, but there appears to be a bug with doing this, so we use a workaround for now.