Move workspace memory-allocation to PyTorch (#661)
* move workspace memory-allocation to PyTorch * refine the code based on the comments * remove unnecessary options * remove bsz from set_seq_len function
Showing
csrc/includes/ds_transformer_cuda.h
100644 → 100755
想要评论请 注册 或 登录