Commit be117fe4 authored by zhushuang's avatar zhushuang
Browse files

fix: disable NVIDIA-dequantize on Iluvatar GPU via ENABLE_NVIDIA_API macro

parent 82b2a84c
#ifdef ENABLE_NVIDIA_API
#include "../../../devices/nvidia/nvidia_handle.cuh"
#include "../../../devices/nvidia/nvidia_kernel_common.cuh"
#include "dequantize_w42f16_kernel.cuh"
......@@ -132,3 +134,5 @@ Descriptor::calculate(
}
} // namespace op::dequantize::nvidia
#endif
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