From 829454a18898d55fb5b1b85737b467f4be07fd3d Mon Sep 17 00:00:00 2001 From: l00800044 Date: Thu, 19 Sep 2024 15:14:35 +0800 Subject: [PATCH] writeback --- torch_npu/meta/_meta_registrations.py | 11 +++++++++++ 1 file changed, 11 insertions(+) diff --git a/torch_npu/meta/_meta_registrations.py b/torch_npu/meta/_meta_registrations.py index 98cc126714..f8a5ab4444 100644 --- a/torch_npu/meta/_meta_registrations.py +++ b/torch_npu/meta/_meta_registrations.py @@ -947,6 +947,17 @@ has_side_effect(torch.ops.npu.npu_prefetch.default) @impl(m, "npu_prefetch") def npu_prefetch_meta(self, dependency, max_size): + torch._check( + max_size > 0, + lambda: f"The max_size should be greater than zero, but got {max_size}.", + ) + + +has_side_effect(torch.ops.npu.npu_writeback.default) + + +@impl(m, "npu_writeback") +def npu_writeback_meta(self, dependency, max_size): torch._check( max_size > 0, lambda: f"The max_size should be greater than zero, but got {max_size}.", -- Gitee