Skip to content

Commit

Permalink
update
Browse files Browse the repository at this point in the history
  • Loading branch information
tangzhiyi11 committed Jan 17, 2024
1 parent f510221 commit 380cd28
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion dicp/dicp/vendor/AscendGraph/codegen/load_and_run.py
Original file line number Diff line number Diff line change
Expand Up @@ -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)
Expand Down

0 comments on commit 380cd28

Please sign in to comment.