xla
Conditionally set default TPU settings in `init.py`
#5696

Merged

Conditionally set default TPU settings in `init.py` #5696

will-cromar merged 3 commits into master from wcromar/tpu-megacore-fix

Set TPU_MEGACORE in configure_topology

93a04d1b

remove

de752256

will-cromar requested a review from

alanwaketan 2 years ago

will-cromar requested a review from

jonb377 2 years ago

JackCaoG approved these changes on 2023-10-10

alanwaketan commented on 2023-10-10

Move back to __init__.py

3b19f155

will-cromar changed the title ~~Set TPU_MEGACORE in configure_topology~~ Conditionally set default TPU settings in `__init__.py` 2 years ago

will-cromar added runtime

alanwaketan approved these changes on 2023-10-10

will-cromar merged a98656d2 into master 2 years ago

Reviewers

alanwaketan

JackCaoG

jonb377

Assignees

No one assigned

Labels

runtime

Milestone

No milestone