Include MKL-DNN header to concat op only when flag is set

test=develop
ce_debug
Michal Gallus 7 years ago
parent f2a880421e
commit 6fdbb365ce

@ -13,11 +13,13 @@ See the License for the specific language governing permissions and
limitations under the License. */
#include "paddle/fluid/operators/concat_op.h"
#include <paddle/fluid/platform/mkldnn_helper.h>
#include <string>
#include <vector>
#ifdef PADDLE_WITH_MKLDNN
#include <paddle/fluid/platform/mkldnn_helper.h>
#endif
namespace paddle {
namespace operators {
using framework::Tensor;

Loading…
Cancel
Save