.. |
__init__.py
|
69d1b9f978
DeepSpeed-Triton for Inference (#3748)
|
1 年之前 |
attention.py
|
69d1b9f978
DeepSpeed-Triton for Inference (#3748)
|
1 年之前 |
gelu.py
|
807d1b5dfc
scripts/check-torchcuda.py: add checking for tensor.is_cuda (#3843)
|
1 年之前 |
layer_norm.py
|
69d1b9f978
DeepSpeed-Triton for Inference (#3748)
|
1 年之前 |
matmul_ext.py
|
bafaf3c0bb
bug fix: triton importing error (#3799)
|
1 年之前 |
mlp.py
|
69d1b9f978
DeepSpeed-Triton for Inference (#3748)
|
1 年之前 |
ops.py
|
69d1b9f978
DeepSpeed-Triton for Inference (#3748)
|
1 年之前 |
residual_add.py
|
807d1b5dfc
scripts/check-torchcuda.py: add checking for tensor.is_cuda (#3843)
|
1 年之前 |
softmax.py
|
69d1b9f978
DeepSpeed-Triton for Inference (#3748)
|
1 年之前 |
triton_matmul_kernel.py
|
69d1b9f978
DeepSpeed-Triton for Inference (#3748)
|
1 年之前 |