fc op use cpu only

7 years ago · d84a1a0010
parent fbc164047d
commit d84a1a0010
1 changed files with 5 additions and 0 deletions
--- a/paddle/fluid/operators/CMakeLists.txt
+++ b/paddle/fluid/operators/CMakeLists.txt
@ -158,6 +158,11 @@ function(op_library TARGET)
        file(APPEND ${pybind_file} "USE_OP_DEVICE_KERNEL(relu, MKLDNN);\n")
      else()
        file(APPEND ${pybind_file} "USE_OP_DEVICE_KERNEL(${TARGET}, MKLDNN);\n")
        # HACK: fc only have cpu kernel
        if (${MKLDNN_FILE} STREQUAL "fc_mkldnn_op")
            file(APPEND ${pybind_file} "USE_CPU_ONLY_OP(${TARGET});\n")
            set(pybind_flag 1)
        endif()
      endif()
    endif()