Skip to content

GitLab

"vscode:/vscode.git/clone" did not exist on "621980bdc0d5a41e224febf962a6e0474e2b14ef"
Commit 07f59c3d authored by wenjh's avatar wenjh
Browse files

Merge branch 'develop_v2.8' into release_v2.8

parents 688c7ab9 a26a0c30
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
transformer_engine/common/gemm/cublaslt_gemm.cu
View file @ 07f59c3d
......@@ -1392,7 +1392,7 @@ void nvte_grouped_gemm(const NVTETensor *A, const NVTETensor *B, NVTETensor *D,
NVTE_ERROR("MOE nvte_grouped_gemm not surpport bias or gelu.");
}
hipblaslt_goupedgemm(inputA, inputB, outputD, m, n, k, b,
hipblaslt_groupedgemm(inputA, inputB, outputD, m, n, k, b,
(transa) ? HIPBLAS_OP_T : HIPBLAS_OP_N,
(transb) ? HIPBLAS_OP_T : HIPBLAS_OP_N,
wspace->data.dptr, wspace->data.shape[0],
......
transformer_engine/common/gemm/rocm_gemm.cu
View file @ 07f59c3d
......@@ -1834,7 +1834,7 @@ class d_userArgsManager {
static userArgsManager UAManager;
static d_userArgsManager d_UAManager;
void hipblaslt_goupedgemm(std::vector<const Tensor*>& inputA, std::vector<const Tensor*>& inputB,
void hipblaslt_groupedgemm(std::vector<const Tensor*>& inputA, std::vector<const Tensor*>& inputB,
std::vector<Tensor*>& outputD, std::vector<int64_t>& m,
std::vector<int64_t>& n, std::vector<int64_t>& k, std::vector<int64_t>& b,
hipblasOperation_t transa, hipblasOperation_t transb, void* workspace,
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment