xla
Set xla_tpu_enable_flash_attention=false to enable libtpu pin update
#8008

Merged

Set xla_tpu_enable_flash_attention=false to enable libtpu pin update #8008

bhavya01 merged 3 commits into master from pallas_fix

Update test_xla_sharding.py to take into account the number of devices

1e4e5d23

fix lint

973f9574

Set xla_tpu_enable_flash_attention=false and update flash attention p…

0a0c42f9

bhavya01 added tpuci

bhavya01 assigned

bhavya01 1 year ago

bhavya01 requested a review from

ManfeiBai 1 year ago

ManfeiBai approved these changes on 2024-09-13

bhavya01 merged 9c7f0835 into master 1 year ago

ManfeiBai added 2024Q3-manfei-review

Reviewers

ManfeiBai

Assignees

bhavya01

Labels

None yet

Milestone

No milestone