diff options
Diffstat (limited to 'src/cpu/operators/CpuMatMul.h')
-rw-r--r-- | src/cpu/operators/CpuMatMul.h | 11 |
1 files changed, 5 insertions, 6 deletions
diff --git a/src/cpu/operators/CpuMatMul.h b/src/cpu/operators/CpuMatMul.h index 24db3da346..2b1b4cf0ff 100644 --- a/src/cpu/operators/CpuMatMul.h +++ b/src/cpu/operators/CpuMatMul.h @@ -21,8 +21,8 @@ * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE * SOFTWARE. */ -#ifndef ACL_SRC_CPU_OPERATORS_CPUMATMUL -#define ACL_SRC_CPU_OPERATORS_CPUMATMUL +#ifndef ACL_SRC_CPU_OPERATORS_CPUMATMUL_H +#define ACL_SRC_CPU_OPERATORS_CPUMATMUL_H #include "arm_compute/core/TensorInfo.h" @@ -93,9 +93,8 @@ public: private: enum InternalTensorIdx { - AsmGemmWorkspace = 0, // Pre-allocate workspace tensors for CpuGemmAssemblyDispatch - PretransposeRHS, // Pre-allocate workspace tensors for CpuGemmAssemblyDispatch - TransposeLHS, + /* Slots 0 - 2 reserved for CpuGemmAssemblyDispatch */ + TransposeLHS = 3, TransposeRHS, Count }; @@ -124,4 +123,4 @@ private: } // namespace cpu } // namespace arm_compute -#endif /* ACL_SRC_CPU_OPERATORS_CPUMATMUL */ +#endif // ACL_SRC_CPU_OPERATORS_CPUMATMUL_H |