Skip to content

GitLab

Commit 4a180009 authored by Pan Zezhong's avatar Pan Zezhong
Browse files

issue/191/fix 修正random sample workspace返回result处理方式

parent 2f47d773
Hide whitespace changes
Inline Side-by-side
Showing with 6 additions and 4 deletions
src/infiniop/ops/random_sample/cuda/random_sample_cuda.cu
View file @ 4a180009
...@@ -26,10 +26,12 @@ infiniStatus_t Descriptor::create( ...@@ -26,10 +26,12 @@ infiniStatus_t Descriptor::create(
auto info = result.take(); auto info = result.take();
size_t workspace_size; size_t workspace_size;
#define CASE_P(CASE, Tidx, Tval) \ #define CASE_P(CASE, Tidx, Tval) \
case CASE: \ case CASE: { \
workspace_size = calculateWorkspace<Tidx, Tval>(info.n); \ auto workspace_result = calculateWorkspace<Tidx, Tval>(info.n); \
break CHECK_RESULT(workspace_result); \
workspace_size = workspace_result.take(); \
} break
#define CASE_I(CASE, Tidx) \ #define CASE_I(CASE, Tidx) \
case CASE: \ case CASE: \
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment