Requirements for using Tensorcores #10791
Unanswered
Bananenbrotesser
asked this question in
Q&A
Replies: 0 comments
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Uh oh!
There was an error while loading. Please reload this page.
-
Hello,
i would like to use the https://github.com/stanfordnqp/jaxwell repo for my FDFD simulations. I want them to be as fast as possible.
The main computational part is the iterative solver for a large system of linear equations. (https://github.com/stanfordnqp/jaxwell/blob/main/jaxwell/cocg.py)
All elements are named tuples (https://github.com/stanfordnqp/jaxwell/blob/main/jaxwell/vecfield.py).
My idea is to modify the code to use the tensorcores of the A100 GPU. Currently tensorboard shows the created kernels as "not eligible for tensorcores"
What requirements do they have? Are there better ideas to achieve a speedup?
Beta Was this translation helpful? Give feedback.
All reactions