Skip to content

Commit

Permalink
get back from CI exper
Browse files Browse the repository at this point in the history
  • Loading branch information
arlesniak committed Jan 14, 2021
1 parent ea2e096 commit 3889476
Showing 1 changed file with 3 additions and 7 deletions.
10 changes: 3 additions & 7 deletions paddle/fluid/framework/operator.cc
Original file line number Diff line number Diff line change
Expand Up @@ -1044,15 +1044,11 @@ bool OperatorWithKernel::SupportsMKLDNN(
const proto::VarType::Type data_type) const {
auto& op_kernels = OperatorWithKernel::AllOpKernels().at(type_);
return std::any_of(op_kernels.begin(), op_kernels.end(),
[](OpKernelMap::const_reference kern_pair) {
// [data_type](OpKernelMap::const_reference
// kern_pair) {
[data_type](OpKernelMap::const_reference kern_pair) {
return platform::is_cpu_place(kern_pair.first.place_) &&
kern_pair.first.library_type_ ==
LibraryType::kMKLDNN;
// &&
// kern_pair.first.data_type_
// == data_type;
LibraryType::kMKLDNN &&
kern_pair.first.data_type_ == data_type;
});
}

Expand Down

0 comments on commit 3889476

Please sign in to comment.