modified workspace

xgqdut2016 · xgqdut2016 · commit c4bdddac2d7d · 2024-11-20T11:00:01.000+08:00
diff --git a/src/ops/random_sample/cuda/random_sample_cuda.cc b/src/ops/random_sample/cuda/random_sample_cuda.cc
@@ -35,7 +35,7 @@ infiniopStatus_t cudaGetRandomSampleWorkspaceSize(RandomSampleCudaDescriptor_t d
     size_t size_scan;
     random_sample_workspace(size_radix_sort, size_scan,
                             desc->voc, desc->dtype);
-    *size = desc->voc * desc->step + std::max(size_radix_sort, size_scan);
+    *size = desc->step + std::max(size_radix_sort, size_scan);
     return STATUS_SUCCESS;
 }
 

Original file line number	Diff line number	Diff line change
`@@ -35,7 +35,7 @@ infiniopStatus_t cudaGetRandomSampleWorkspaceSize(RandomSampleCudaDescriptor_t d`
`35`	`35`	`size_t size_scan;`
`36`	`36`	`random_sample_workspace(size_radix_sort, size_scan,`
`37`	`37`	`desc->voc, desc->dtype);`
`38`		`- size = desc->voc desc->step + std::max(size_radix_sort, size_scan);`
	`38`	`+ *size = desc->step + std::max(size_radix_sort, size_scan);`
`39`	`39`	`return STATUS_SUCCESS;`
`40`	`40`	`}`
`41`	`41`