You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
jax.lax.dot_general has a new algorithm option for precisely specifying the precision, which should be supported in Pallas. Also support the simulated precision modes on GPU, such as BF16_BF16_F32_x6 and TF32_TF32_F32_x3 (already supported by triton.language.dot via the input_precision="tf32x3" option).
jax.lax.dot_general
has a newalgorithm
option for precisely specifying the precision, which should be supported in Pallas. Also support the simulated precision modes on GPU, such asBF16_BF16_F32_x6
andTF32_TF32_F32_x3
(already supported bytriton.language.dot
via theinput_precision="tf32x3"
option).@chr1sj0nes, @sharadmv, @dfm
The text was updated successfully, but these errors were encountered: