Lucas Wilkinson aeb37c2a72

[CI/Build] Per file CUDA Archs (improve wheel size and dev build times) (#8845)

2024-10-03 22:55:25 -04:00

..

marlin_kernels

[Kernel] Split Marlin MoE kernels into multiple files (#8661)

2024-09-24 09:31:42 -07:00

marlin_moe_ops.cu

[CI/Build] Per file CUDA Archs (improve wheel size and dev build times) (#8845)

2024-10-03 22:55:25 -04:00

moe_ops.h

[Kernel][Misc] Use TORCH_LIBRARY instead of PYBIND11_MODULE for custom ops (#5047)

2024-06-09 16:23:30 -04:00

topk_softmax_kernels.cu

[Kernel][Misc] Use TORCH_LIBRARY instead of PYBIND11_MODULE for custom ops (#5047)

2024-06-09 16:23:30 -04:00

torch_bindings.cpp

[CI/Build] Per file CUDA Archs (improve wheel size and dev build times) (#8845)

2024-10-03 22:55:25 -04:00