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