jax
7483a3eb - layout: Implement out-of-line template method as a non-template.

Commit

173 days ago

layout: Implement out-of-line template method as a non-template. Template functions need to be defined inline or have all the expected instantiations specified in the header. With only three instantations (and ostreams leading to significant code being generated due to the huge number of lookups for operator<< overloads), it's probably easiest to define overloads that call a templated static out-of-line. Discovered by building with thinlto, which more strictly applies C++ rules on such things. PiperOrigin-RevId: 784324779

References

#30285 - layout: Implement out-of-line template method as a non-template.

#31381 - Remove old ROCm build code

#31720 - Fix ann_test.py numerical bug in target reshape

#31768 - [ROCm] Support lowering through PJRT_Triton_Extension

#32115 - Relax version requirements for ROCm Jax Plugin wheels

#33157 - Resolve undefined behavior in bitshift unit test

#33186 - Make nvidia version data optional for ROCm builds

#579 - Create rocm-test-requirements.txt

#580 - GESVDJ support for ROCm GPUs in JAX

#581 - Fix/pallas tests shared memory

#584 - Use plain bazel to test jax, use hermetic rocm dependency