add some comment

revert-12469-sum_op_dim_fix
tensor-tang 7 years ago
parent 5377edd282
commit 171a0e2b42

@ -276,6 +276,7 @@ class GRUCPUKernel : public framework::OpKernel<T> {
context.Attr<std::string>("gate_activation")); context.Attr<std::string>("gate_activation"));
#ifdef PADDLE_WITH_MKLML #ifdef PADDLE_WITH_MKLML
// use MKL packed to speedup GEMM
if (FLAGS_paddle_num_threads >= 4) { if (FLAGS_paddle_num_threads >= 4) {
auto blas = math::GetBlas<DeviceContext, T>(dev_ctx); auto blas = math::GetBlas<DeviceContext, T>(dev_ctx);
T* packed_gate = blas.GEMM_ALLOC(CblasBMatrix, 1 /*height of C*/, T* packed_gate = blas.GEMM_ALLOC(CblasBMatrix, 1 /*height of C*/,

Loading…
Cancel
Save