提交 ed10d44b 编写于 作者: M Megvii Engine Team

feat(opencl/bazel): support bazel build with ocl only

how to use: bazel xxxxx --define enable_opencl_build_only=1

GitOrigin-RevId: 8badb8dc1306a05a73a465321e34f212a74620ae
上级 8efd02d8
...@@ -2,7 +2,9 @@ ...@@ -2,7 +2,9 @@
#include "src/common/handle_impl.h" #include "src/common/handle_impl.h"
#include "src/common/utils.h" #include "src/common/utils.h"
#if !MGE_BUILD_CL_ONLY
#include "src/fallback/handle.h" #include "src/fallback/handle.h"
#endif
#include "src/naive/handle.h" #include "src/naive/handle.h"
#include "midout.h" #include "midout.h"
......
#pragma once #pragma once
#include "megdnn/oprs.h" #include "megdnn/oprs.h"
#if !MGE_BUILD_CL_ONLY
#include "src/fallback/conv_bias/opr_impl.h" #include "src/fallback/conv_bias/opr_impl.h"
#endif
namespace megdnn { namespace megdnn {
#if MGE_BUILD_CL_ONLY
using NonlineMode = ConvBias::Param::NonlineMode;
using BiasMode = ConvBiasForward::BiasMode;
#endif
namespace { namespace {
enum NchwNchwxxType { enum NchwNchwxxType {
NCHW44_FP32, NCHW44_FP32,
...@@ -173,4 +181,4 @@ inline bool nchw_nchwxx_valid<NCHW88_FP16>( ...@@ -173,4 +181,4 @@ inline bool nchw_nchwxx_valid<NCHW88_FP16>(
#endif #endif
} // namespace } // namespace
} // namespace megdnn } // namespace megdnn
\ No newline at end of file
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册