Fix uninitialized variable bug in svd plan

NVIDIA · Dec 19, 2024 · 4ec8ab5 · 4ec8ab5
1 parent 0770c7b
commit 4ec8ab5
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/include/matx/transforms/svd/svd_cuda.h b/include/matx/transforms/svd/svd_cuda.h
@@ -668,7 +668,7 @@ class matxDnSVDCUDAPlan_t : matxDnCUDASolver_t {
     cusolverStatus_t ret;
 
     // Use all mode for a larger workspace size that works for all modes
-    if (!batched) {
+    if (params.method == SVDMethod::GESVD) {
       ret = cusolverDnXgesvd_bufferSize(
               this->handle, this->dn_params, 'A', 'A', params.m, params.n,
               MatXTypeToCudaType<T1>(), params.A, params.m,
@@ -874,7 +874,6 @@ class matxDnSVDCUDAPlan_t : matxDnCUDASolver_t {
   std::vector<T4 *> batch_vt_ptrs;
   gesvdjInfo_t batch_params = nullptr;
   DnSVDCUDAParams_t params;
-  bool batched;
 };
 
 /**