Fix fc mkldnn op

revert-12383-port_py3_syntax
yuyang18 7 years ago
parent 82866d4a18
commit 6038a63120
No known key found for this signature in database
GPG Key ID: 6DFF29878217BE5F

@ -115,6 +115,7 @@ class MKLDNNMemory {
template <typename T>
class FCMKLDNNOpKernel : public paddle::framework::OpKernel<T> {
public:
void Compute(const paddle::framework::ExecutionContext& ctx) const override {
PADDLE_ENFORCE(paddle::platform::is_cpu_place(ctx.GetPlace()),
"It must use CPUPlace.");

Loading…
Cancel
Save