From c9afce1a052163f6b3c985c82eb27b72bb6ee336 Mon Sep 17 00:00:00 2001 From: l00800044 Date: Fri, 1 Mar 2024 15:21:08 +0800 Subject: [PATCH] groupmm add --- torch_npu/meta/meta_registrations.py | 7 +++++++ 1 file changed, 7 insertions(+) diff --git a/torch_npu/meta/meta_registrations.py b/torch_npu/meta/meta_registrations.py index 1458210ffe..dbfbc79dd7 100644 --- a/torch_npu/meta/meta_registrations.py +++ b/torch_npu/meta/meta_registrations.py @@ -461,3 +461,10 @@ def npu_anti_quant_meta(x, scale, *, offset=None, dst_dtype=None, src_dtype=None if dst_dtype is None: return torch.empty_like(x, dtype=torch.float16) return torch.empty_like(x, dtype=dst_dtype) + +@impl(m, "npu_grouped_matmul") +def npu_grouped_matmul_meta(x, weight, *, bias=None, group_list=None, split_item=0): + empty_list = [] + for item in x: + empty_list.append(torch.empty_like(item)) + return empty_list \ No newline at end of file -- Gitee