Browse Source

build(mge): clang compile and link error

GitOrigin-RevId: 7062b451aa
tags/v0.4.0
Megvii Engine Team Xinran Xu 5 years ago
parent
commit
e3de2011c0
2 changed files with 1 additions and 7 deletions
  1. +0
    -2
      dnn/src/cuda/deformable_conv/opr_impl.cpp
  2. +1
    -5
      src/CMakeLists.txt

+ 0
- 2
dnn/src/cuda/deformable_conv/opr_impl.cpp View File

@@ -141,7 +141,6 @@ AlgoBwdFlt* BwdFlt::get_algorithm_heuristic(
size_t BwdFlt::get_workspace_in_bytes( size_t BwdFlt::get_workspace_in_bytes(
const TensorLayout& im, const TensorLayout& offset, const TensorLayout& mask, const TensorLayout& im, const TensorLayout& offset, const TensorLayout& mask,
const TensorLayout& out_grad, const TensorLayout& filter_grad) { const TensorLayout& out_grad, const TensorLayout& filter_grad) {
AlgoBase::SizeArgs args();
auto algo = get_algorithm(this, im, offset, mask, out_grad, filter_grad); auto algo = get_algorithm(this, im, offset, mask, out_grad, filter_grad);
return algo->get_workspace_in_bytes({this, im, offset, mask, out_grad, filter_grad}); return algo->get_workspace_in_bytes({this, im, offset, mask, out_grad, filter_grad});
} }
@@ -207,7 +206,6 @@ size_t BwdData::get_workspace_in_bytes(
const TensorLayout& offset, const TensorLayout& mask, const TensorLayout& offset, const TensorLayout& mask,
const TensorLayout& out_grad, const TensorLayout& im_grad, const TensorLayout& out_grad, const TensorLayout& im_grad,
const TensorLayout& offset_grad, const TensorLayout& mask_grad) { const TensorLayout& offset_grad, const TensorLayout& mask_grad) {
AlgoBase::SizeArgs args();
auto algo = get_algorithm(this, im, filter, offset, mask, out_grad, auto algo = get_algorithm(this, im, filter, offset, mask, out_grad,
im_grad, offset_grad, mask_grad); im_grad, offset_grad, mask_grad);
return algo->get_workspace_in_bytes({this, im, filter, offset, mask, out_grad, return algo->get_workspace_in_bytes({this, im, filter, offset, mask, out_grad,


+ 1
- 5
src/CMakeLists.txt View File

@@ -130,11 +130,7 @@ if (MGB_WITH_FLATBUFFERS)
target_include_directories(megbrain PRIVATE ${GENERATED_FLATBUFFERS_CONVERTER_PATH}) target_include_directories(megbrain PRIVATE ${GENERATED_FLATBUFFERS_CONVERTER_PATH})
endif() endif()
if(UNIX) if(UNIX)
target_link_libraries(megbrain dl rt)
find_library(libatomic atomic)
if(NOT libatomic STREQUAL "libatomic-NOTFOUND")
target_link_libraries(megbrain ${libatomic})
endif()
target_link_libraries(megbrain dl rt atomic)
endif() endif()


add_library(megengine SHARED $<TARGET_PROPERTY:megbrain,SOURCES>) add_library(megengine SHARED $<TARGET_PROPERTY:megbrain,SOURCES>)


Loading…
Cancel
Save