Skip to content

GitLab

Commit d790f7b4 authored by wooway777's avatar wooway777
Browse files

issue/1033 - fix format

parent b1ee0a8a
Show whitespace changes
Inline Side-by-side
Showing with 41 additions and 41 deletions
src/infiniop/ops/random_sample/nvidia/random_sample_kernel.cuh
View file @ d790f7b4
......@@ -18,8 +18,8 @@ static cudaError argMax_(
cudaStream_t stream) {
#if CUDART_VERSION >= 11000
// New interface: separate value and index outputs
T* max_value = &kv_pair->value;
int* max_index = &kv_pair->key;
T *max_value = &kv_pair->value;
int *max_index = &kv_pair->key;
return cub::DeviceReduce::ArgMax(
workspace_ptr, workspace_len,
logits, max_value, max_index, n,
......
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