Commit 05d05fd7 authored by wooway777's avatar wooway777
Browse files

issue/1033 - fix argmax compile on hygon and qy

parent e7c743ab
......@@ -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;
......
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