search

Dao-AILab / causal-conv1d

Causal depthwise conv1d in CUDA, with a PyTorch interface
BSD 3-Clause "New" or "Revised" License
329 stars 61 forks source link

TORCH_CUDA_ARCH_LIST support #39

Open bhack opened 3 hours ago

bhack commented 3 hours ago

Can we support arch list specification with this env as documented in https://pytorch.org/docs/stable/cpp_extension.html ?

bhack commented 2 hours ago

E.g. https://github.com/facebookresearch/xformers/blob/main/setup.py#L271-L296