From dae4652c6dce8b8a52af491259d3570af785821a Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E6=B1=9F=E4=BF=8A=E6=88=90?= Date: Tue, 17 Dec 2024 11:30:30 +0800 Subject: [PATCH] aGlobal bGlobal add address_ --- impl/matmul/matmul_impl.h | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/impl/matmul/matmul_impl.h b/impl/matmul/matmul_impl.h index 26eb8368..d178d39c 100644 --- a/impl/matmul/matmul_impl.h +++ b/impl/matmul/matmul_impl.h @@ -6040,9 +6040,9 @@ __aicore__ inline void MatmulImplBase 0 && batchB_ > 0 && (batchA_ % batchB_ == 0 || batchB_ % batchA_ == 0)); GlobalTensor aGlobal; - aGlobal.SetGlobalBuffer(var.aGlobal_); + aGlobal.SetGlobalBuffer(var.aGlobal_.address_); GlobalTensor bGlobal; - bGlobal.SetGlobalBuffer(var.bGlobal_); + bGlobal.SetGlobalBuffer(var.bGlobal_.address_); GlobalTensor biasGlobal; if constexpr (ToMatmulConfig(MM_CFG).enableSetBias) { if (var.enableBias_) { @@ -6289,9 +6289,9 @@ __aicore__ inline void MatmulImplBase 0 && batchB_ > 0 && (batchA_ % batchB_ == 0 || batchB_ % batchA_ == 0)); GlobalTensor aGlobal; - aGlobal.SetGlobalBuffer(var.aGlobal_); + aGlobal.SetGlobalBuffer(var.aGlobal_.address_); GlobalTensor bGlobal; - bGlobal.SetGlobalBuffer(var.bGlobal_); + bGlobal.SetGlobalBuffer(var.bGlobal_.address_); GlobalTensor biasGlobal; if constexpr (ToMatmulConfig(MM_CFG).enableSetBias) { if (var.enableBias_) { -- Gitee