Update ptxla training #9864
yiyixuxu
merged
20 commits into
huggingface:main
from
entrpn:update_ptxla_training
update ptxla example
f04ee1d1
update ptxla example based on Pei's comments.
96af06e8
add print loss cli argument. Run make style and quality.
6234a37b
make measure_start_step an argument.
b5161343
use PORT variable across the script.
8c47f35d
Merge branch 'main' into update_ptxla_training
10b6ba16
split out xla flash attention from base AttnProcessor
8b3cbb16
Merge branch 'main' into update_ptxla_training
6c74c799
use version check for torch_xla
2c00cbd0
Merge branch 'main' into update_ptxla_training
5969ce40
setup the option to use xla flash attention or not
fb29e372
Merge branch 'main' into update_ptxla_training
46baa568
naming nit
df31c9df
Merge branch 'main' into update_ptxla_training
f1ade2e5
format fix with ruff cmd
ff332e68
adding warning message
dbe4725f
format fix with ruff cmd
c012fafe
Merge branch 'main' into update_ptxla_training
0225ed75
make style
03089f51
Merge branch 'main' into update_ptxla_training
719fcf9f
yiyixuxu
approved these changes
on 2024-12-06
Login to write a write a comment.
Login via GitHub