| .. |
|
all_reduce
|
156f577f79
feat: switch from `PYBIND11_MODULE` to `TORCH_LIBRARY` (#569)
|
hai 7 meses |
|
attention
|
156f577f79
feat: switch from `PYBIND11_MODULE` to `TORCH_LIBRARY` (#569)
|
hai 7 meses |
|
backup
|
f8dfac6372
chore: attention refactor and upstream sync apr01 (#365)
|
hai 11 meses |
|
cpu
|
271a680026
feat: inference support for PowerPC ISA
|
hai 7 meses |
|
hadamard
|
5d288aa76c
feat: add fast hadamard transformation kernels (#232)
|
hai 1 ano |
|
moe
|
156f577f79
feat: switch from `PYBIND11_MODULE` to `TORCH_LIBRARY` (#569)
|
hai 7 meses |
|
punica
|
4f42985b5c
feat: qwen2 lora shapes
|
hai 7 meses |
|
quantization
|
b03b4d4c8c
fix: compute cutlass 3.x epilogues in fp32 instead of 16
|
hai 7 meses |
|
activation_kernels.cu
|
c0c336aaa3
refactor: registry for processing model inputs; quick_gelu; clip model support
|
hai 7 meses |
|
cache.h
|
156f577f79
feat: switch from `PYBIND11_MODULE` to `TORCH_LIBRARY` (#569)
|
hai 7 meses |
|
cache_kernels.cu
|
156f577f79
feat: switch from `PYBIND11_MODULE` to `TORCH_LIBRARY` (#569)
|
hai 7 meses |
|
cuda_compat.h
|
00acf371f9
rocm: fused topk softmax
|
hai 7 meses |
|
cuda_utils.h
|
156f577f79
feat: switch from `PYBIND11_MODULE` to `TORCH_LIBRARY` (#569)
|
hai 7 meses |
|
cuda_utils_kernels.cu
|
156f577f79
feat: switch from `PYBIND11_MODULE` to `TORCH_LIBRARY` (#569)
|
hai 7 meses |
|
dispatch_utils.h
|
156f577f79
feat: switch from `PYBIND11_MODULE` to `TORCH_LIBRARY` (#569)
|
hai 7 meses |
|
layernorm_kernels.cu
|
156f577f79
feat: switch from `PYBIND11_MODULE` to `TORCH_LIBRARY` (#569)
|
hai 7 meses |
|
ops.h
|
c0c336aaa3
refactor: registry for processing model inputs; quick_gelu; clip model support
|
hai 7 meses |
|
pos_encoding_kernels.cu
|
156f577f79
feat: switch from `PYBIND11_MODULE` to `TORCH_LIBRARY` (#569)
|
hai 7 meses |
|
reduction.cuh
|
aba03b4756
feat: dynamic per-token activation quantization
|
hai 7 meses |
|
registration.h
|
156f577f79
feat: switch from `PYBIND11_MODULE` to `TORCH_LIBRARY` (#569)
|
hai 7 meses |
|
torch_bindings.cpp
|
c0c336aaa3
refactor: registry for processing model inputs; quick_gelu; clip model support
|
hai 7 meses |
AlpinDale
b03b4d4c8c
fix: compute cutlass 3.x epilogues in fp32 instead of 16