Skip to content

GitLab

Commit be117fe4 authored by zhushuang's avatar zhushuang
Browse files

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

parent 82b2a84c
Show whitespace changes
Inline Side-by-side
Showing with 5 additions and 1 deletion
src/infiniop/ops/dequantize/nvidia/dequantize_w42f16_nvidia.cu
View file @ be117fe4
#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