Tri Dao d380e87fb6 Don't use Smem_dp_sum in backward pass ... To reduce smem usage for SM75		2022-06-04 16:01:36 -07:00
..
flash_attn	Don't use Smem_dp_sum in backward pass	2022-06-04 16:01:36 -07:00