Add opaque pointer types for BLAS/SOLVER handle pool separation

Addresses handle pool singleton sharing issue between different GPU
operation types in ROCm/HIP backend.

Changes made:
- Add opaque pointer typedefs for HIP backend:
  * typedef struct hipblasHandle_* gpublasHandle_t
  * typedef struct hipsolverHandle_* gpusolverDnHandle_t
- Implement inline wrapper functions for type-safe handle operations:
  * gpublasCreate() and gpublasSetStream() for BLAS handles
  * gpusolverDnCreate() and gpusolverDnSetStream() for SOLVER handles

Author: phambinhfin

jaxlib/gpu/vendor.h

@@ -431,10 +431,12 @@ typedef hipDoubleComplex gpuDoubleComplex;
 typedef hipComplex gpublasComplex;
 typedef hipDoubleComplex gpublasDoubleComplex;
 
-typedef hipsolverHandle_t gpusolverDnHandle_t;
+// Create unique opaque pointer types for proper singleton separation - BLAS and SOLVER only
+typedef struct hipblasHandle_* gpublasHandle_t;
+typedef struct hipsolverHandle_* gpusolverDnHandle_t;
+
 typedef hipblasFillMode_t gpublasFillMode_t;
 typedef hipsolverFillMode_t gpusolverFillMode_t;
-typedef hipblasHandle_t gpublasHandle_t;
 typedef hipblasOperation_t gpublasOperation_t;
 typedef hipblasStatus_t gpublasStatus_t;
 typedef hipCtx_t gpuContext_t;
@@ -480,8 +482,15 @@ typedef hipsparseDnVecDescr_t gpusparseDnVecDescr_t;
 #define GPU_C_64F HIP_C_64F
 #define GPU_R_64F HIP_R_64F
 
-#define gpublasCreate hipblasCreate
-#define gpublasSetStream hipblasSetStream
+// Inline wrapper functions for BLAS handles to ensure unique types
+inline hipblasStatus_t gpublasCreate(gpublasHandle_t* handle) {
+  return hipblasCreate(reinterpret_cast<hipblasHandle_t*>(handle));
+}
+
+inline hipblasStatus_t gpublasSetStream(gpublasHandle_t handle, hipStream_t stream) {
+  return hipblasSetStream(reinterpret_cast<hipblasHandle_t>(handle), stream);
+}
+
 #define gpublasSgeqrfBatched hipblasSgeqrfBatched
 #define gpublasDgeqrfBatched hipblasDgeqrfBatched
 #define gpublasCgeqrfBatched hipblasCgeqrfBatched
@@ -531,8 +540,15 @@ typedef hipsparseDnVecDescr_t gpusparseDnVecDescr_t;
 #define GPUDNN_LSTM miopenLSTM
 #define GPUDNN_BIDIRECTIONAL miopenRNNbidirection
 
-#define gpusolverDnCreate hipsolverCreate
-#define gpusolverDnSetStream hipsolverSetStream
+// Inline wrapper functions for SOLVER handles to ensure unique types
+inline hipsolverStatus_t gpusolverDnCreate(gpusolverDnHandle_t* handle) {
+  return hipsolverCreate(reinterpret_cast<hipsolverHandle_t*>(handle));
+}
+
+inline hipsolverStatus_t gpusolverDnSetStream(gpusolverDnHandle_t handle, hipStream_t stream) {
+  return hipsolverSetStream(reinterpret_cast<hipsolverHandle_t>(handle), stream);
+}
+
 #define gpusolverDnCreateSyevjInfo hipsolverCreateSyevjInfo
 #define gpusolverDnDestroySyevjInfo hipsolverDestroySyevjInfo
 #define gpusolverDnSgeqrf hipsolverSgeqrf