feat(2D): support selective-checkpoint-offload for 2D attention #1258
Job | Run time |
---|---|
2m 4s | |
1m 40s | |
1m 44s | |
1m 42s | |
7m 13s | |
10m 15s | |
2m 0s | |
1m 43s | |
1m 49s | |
1m 45s | |
1m 55s | |
33m 50s |
Job | Run time |
---|---|
2m 4s | |
1m 40s | |
1m 44s | |
1m 42s | |
7m 13s | |
10m 15s | |
2m 0s | |
1m 43s | |
1m 49s | |
1m 45s | |
1m 55s | |
33m 50s |