- 07 9月, 2022 1 次提交
-
-
由 Wilber 提交于
-
- 01 8月, 2022 1 次提交
-
-
由 Leo Chen 提交于
* remove cudaDeviceContext * remove more template * fix rocm compile * remove alias name CUDADeviceContext * fix compile * fix tests * revert changes
-
- 30 6月, 2022 1 次提交
-
-
由 wanghuancoder 提交于
* fused_gate_attention manual code in eager * refine * refine * refine * refine * refine * refine
-
- 26 6月, 2022 1 次提交
-
-
由 Sing_chan 提交于
-
- 15 6月, 2022 1 次提交
-
-
由 Yiqun Liu 提交于
* Optimize prod's python implementation for dygraph. * Change key_dim to head_dim. * Add comment in unittest. * Disable TF32 in unittest.
-
- 08 6月, 2022 1 次提交
-
-
由 Yiqun Liu 提交于
* Polish codes and memory usage for fused_gate_attention. * Fix wrong reduce_dims in fused_gate_attention when computing gradient of nonbatched_bias.
-
- 05 6月, 2022 1 次提交
-
-
由 Sing_chan 提交于
-
- 30 5月, 2022 1 次提交
-
-
由 crystal 提交于
-