support dynamic sequence length in transformer kernels (#424)
Co-authored-by: NConglong Li <conglong.li@gmail.com>
Showing
csrc/includes/context.h
100644 → 100755
csrc/includes/ds_transformer_cuda.h
100755 → 100644
csrc/includes/softmax.h
100644 → 100755
csrc/transformer/ds_transformer_cuda.cpp
100755 → 100644
此差异已折叠。
csrc/transformer/gelu_kernels.cu
100755 → 100644
csrc/transformer/normalize_kernels.cu
100755 → 100644
此差异已折叠。
想要评论请 注册 或 登录