add parameter explanation for scatter_sms

rapidsai · Feb 1, 2024 · 11f5922 · 11f5922
1 parent d3316e8
commit 11f5922
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/cpp/include/wholememory/wholememory_op.h b/cpp/include/wholememory/wholememory_op.h
@@ -55,7 +55,7 @@ wholememory_error_code_t wholememory_scatter(wholememory_tensor_t input_tensor,
                                              wholememory_tensor_t wholememory_tensor,
                                              wholememory_env_func_t* p_env_fns,
                                              void* stream,
-                                             int gather_sms = -1);
+                                             int scatter_sms = -1);
 
 /**
  * Just a test function,