[Jax][Pallas][Mosaic] Implement platform dependent diag, with branch selection driven by constant prop in mosaic lowering. #69147
Job | Run time |
---|---|
10m 42s | |
6m 49s | |
1m 28s | |
3m 2s | |
2m 21s | |
1m 14s | |
1m 13s | |
26m 49s |
Job | Run time |
---|---|
10m 42s | |
6m 49s | |
1m 28s | |
3m 2s | |
2m 21s | |
1m 14s | |
1m 13s | |
26m 49s |