Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
jerrrrry
infinicore
Commits
05d05fd7
Commit
05d05fd7
authored
Mar 06, 2026
by
wooway777
Browse files
issue/1033 - fix argmax compile on hygon and qy
parent
e7c743ab
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
1 addition
and
1 deletion
+1
-1
src/infiniop/ops/random_sample/nvidia/random_sample_kernel.cuh
...nfiniop/ops/random_sample/nvidia/random_sample_kernel.cuh
+1
-1
No files found.
src/infiniop/ops/random_sample/nvidia/random_sample_kernel.cuh
View file @
05d05fd7
...
...
@@ -16,7 +16,7 @@ static cudaError argMax_(
void
*
workspace_ptr
,
size_t
&
workspace_len
,
cudaStream_t
stream
)
{
#if CUDART_VERSION >= 11000
#if CUDART_VERSION >= 11000
&& !defined(ENABLE_QY_API) && !defined(ENABLE_HYGON_API)
// New interface: separate value and index outputs
T
*
max_value
=
&
kv_pair
->
value
;
int
*
max_index
=
&
kv_pair
->
key
;
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment