From 380cd2817abe07d088c5b7171246d3abe5856d97 Mon Sep 17 00:00:00 2001 From: tangzhiyi11 Date: Wed, 17 Jan 2024 18:24:03 +0800 Subject: [PATCH] update --- dicp/dicp/vendor/AscendGraph/codegen/load_and_run.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/dicp/dicp/vendor/AscendGraph/codegen/load_and_run.py b/dicp/dicp/vendor/AscendGraph/codegen/load_and_run.py index 5b0a5ea2c0..68a37f4f79 100644 --- a/dicp/dicp/vendor/AscendGraph/codegen/load_and_run.py +++ b/dicp/dicp/vendor/AscendGraph/codegen/load_and_run.py @@ -121,7 +121,7 @@ def __init__(self): def init_work_weight_ptr(self): if self.work_ptr is None: - self.work_size = 18 * 1024 * 1024 * 1024 + self.work_size = 15 * 1024 * 1024 * 1024 self.work_ptr, ret = acl.rt.malloc(self.work_size, ACL_MEM_MALLOC_HUGE_FIRST) check_ret("acl.rt.malloc", ret)