wip titan x bug

ginkgo-project · Aug 13, 2024 · dc85ff5 · dc85ff5
1 parent ca74968
commit dc85ff5
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/cuda/solver/batch_bicgstab_kernels.cu b/cuda/solver/batch_bicgstab_kernels.cu
@@ -157,9 +157,12 @@ public:
  value_type>(
  shmem_per_blk, padded_num_rows, mat.get_single_item_num_nnz(),
  b.num_rhs);
- const size_t shared_size =
- sconf.n_shared * padded_num_rows * sizeof(value_type) +
- (sconf.prec_shared ? prec_size : 0);
+ // const size_t shared_size =
+ // sconf.n_shared * padded_num_rows * sizeof(value_type) +
+ // (sconf.prec_shared ? prec_size : 0);
+ const size_t shared_size = 1200;
+ // sconf.n_shared * padded_num_rows * sizeof(value_type) +
+ // (sconf.prec_shared ? prec_size : 0);
  auto workspace = gko::array<value_type>(
  exec_,
  sconf.gmem_stride_bytes * num_batch_items / sizeof(value_type));