Sytrd solver and SytrdDescriptor should NOT be CUDA only

thisiscam · Nov 11, 2022 · 31d8f62 · 31d8f62
1 parent c359c79
commit 31d8f62
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/jaxlib/gpu/solver_kernels.h b/jaxlib/gpu/solver_kernels.h
@@ -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