Skip to content

GitLab

"git@developer.sourcefind.cn:yaoyuping/nndetection.git" did not exist on "1235f7372108dfb0b85c6a4aa08613fcf5bf93f3"
Unverified Commit d3d982df authored by PanZezhong1725's avatar PanZezhong1725 Committed by GitHub
Browse files

Merge pull request #470 from InfiniTensor/issue/469 

issue/469: disable NVIDIA-dequantize on Iluvatar GPU via ENABLE_NVIDIA_API marco
parents 82b2a84c be117fe4
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 @ d3d982df
#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