diff --git a/torch_npu/meta/_meta_registrations.py b/torch_npu/meta/_meta_registrations.py index 98cc1267149bc2248f9111eee103b6a0623175b9..f8a5ab4444ee604dd1046c63d77c7a3aa691bd96 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}.",