This repository was archived by the owner on Aug 11, 2020. It is now read-only.
Repository navigation
This repository was archived by the owner on Aug 11, 2020. It is now read-only.
dot_engine-inl.h MSHADOW_STAND_ALONE BLASEngine<cpu, float>::gemm error #329
Description
Hi!
The method BLASEngine<cpu,float>::gemm in MSHADOW_STAND_ALONE branch
contains the error in the shape of dst tensor(dot_engine-inl.h).
It is written
Tensor<cpu, 2, float> lhs((float*)B, Shape2(transpose_left ? k : n, transpose_left ? n : k)); // NOLINT()
Tensor<cpu, 2, float> rhs((float)A, Shape2(transpose_right ? m : k, transpose_right ? k : m)); // NOLINT(*)
Tensor<cpu, 2, float> dst(C, Shape2(m, n)); <-- (wrong shape)
Must be
Tensor<cpu, 2, float> dst(C, Shape2(n, m));
This error occurs when executing operator 'dot' and 'batch_dot' in JavaScript
Activity
szha commentedon Aug 4, 2019
This code base has been donated to the Apache MXNet project per #373, and repo is deprecated. Future development and issue tracking should continue in Apache MXNet.