"...git@developer.sourcefind.cn:tsoc/superbenchmark.git" did not exist on "0fdfe4c3116cf6f8e1d3f150200ecba3856e0d2b"
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_handle.cuh"
#include "../../../devices/nvidia/nvidia_kernel_common.cuh" #include "../../../devices/nvidia/nvidia_kernel_common.cuh"
#include "dequantize_w42f16_kernel.cuh" #include "dequantize_w42f16_kernel.cuh"
...@@ -132,3 +134,5 @@ Descriptor::calculate( ...@@ -132,3 +134,5 @@ Descriptor::calculate(
} }
} // namespace op::dequantize::nvidia } // 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