Skip to content
This repository has been archived by the owner on Mar 1, 2023. It is now read-only.

Make climate models strong scale on TPUs #2063

Open
bischtob opened this issue Feb 25, 2021 · 5 comments
Open

Make climate models strong scale on TPUs #2063

bischtob opened this issue Feb 25, 2021 · 5 comments
Assignees
@vchuravy
Labels
good first issue Good for newcomers help wanted Extra attention is needed

Comments

@bischtob
Copy link
Contributor

Description

Self-explanatory. Not high priority now :).
@bischtob bischtob added the good first issue Good for newcomers label Feb 25, 2021
@vchuravy vchuravy added the help wanted Extra attention is needed label Feb 26, 2021
@blallen
Copy link
Contributor

blallen commented Mar 19, 2021

Can we run at half-precision and get reasonable results? feel like that might be a first step

@jkozdon
Copy link

jkozdon commented Mar 19, 2021

Can we run at half-precision and get reasonable results? feel like that might be a first step

I would start with making sure that Float32 gives the right results. (Not sure it does)

@jakebolewski
Copy link
Contributor

There are a lot of steps to make this even remotely feasible lol

@bischtob
Copy link
Contributor Author

I hope everyone understands that this was a joke ;)

@ali-ramadhan
Copy link
Member

Not sure if 100% joke (I was also interested in trying out TPUs at some point) but I think XLA.jl (Julia on TPUs) is dead for now: JuliaGPU/XLA.jl#45 (comment)

Maybe XLA.jl is one of the many steps @jakebolewski alluded to...

Sign up for free to subscribe to this conversation on GitHub. Already have an account? Sign in.
Assignees

@vchuravy vchuravy

Labels
good first issue Good for newcomers help wanted Extra attention is needed
Projects
None yet
Milestone
No milestone
Development

No branches or pull requests
6 participants
@vchuravy @jakebolewski @jkozdon @blallen @bischtob @ali-ramadhan