mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-16 11:56:07 +00:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #13215 from ROCmSoftwarePlatform:sytrd

Browse Source 
PiperOrigin-RevId: 487940256
This commit is contained in:
jax authors 2022-11-11 16:11:27 -08:00
commit d4cc5882d2
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
jaxlib/gpu/solver_kernels.h
View File

@ -160,6 +160,7 @@ struct GesvdjDescriptor {
void Gesvdj(gpuStream_t stream, void** buffers, const char* opaque,
size_t opaque_len, XlaCustomCallStatus* status);
#endif // JAX_GPU_CUDA
// sytrd/hetrd: Reduction of a symmetric (Hermitian) matrix to tridiagonal form.
struct SytrdDescriptor {
@ -171,7 +172,6 @@ struct SytrdDescriptor {
void Sytrd(gpuStream_t stream, void** buffers, const char* opaque,
size_t opaque_len, XlaCustomCallStatus* status);
#endif // JAX_GPU_CUDA
} // namespace JAX_GPU_NAMESPACE
} // namespace jax