Skip to content

Commit

Permalink
Merge pull request cupy#8942 from 99991/main
Browse files Browse the repository at this point in the history
Fix typo in _cretate_frame_tree
  • Loading branch information
kmaehashi authored and chainer-ci committed Feb 12, 2025
1 parent 1e4a78b commit 5f6b597
Showing 1 changed file with 3 additions and 3 deletions.
6 changes: 3 additions & 3 deletions cupy/cuda/memory_hooks/line_profile.py
Original file line number Diff line number Diff line change
Expand Up @@ -54,13 +54,13 @@ def __init__(self, max_depth=0):

# callback
def malloc_preprocess(self, device_id, size, mem_size):
self._cretate_frame_tree(used_bytes=mem_size)
self._create_frame_tree(used_bytes=mem_size)

# callback
def alloc_preprocess(self, device_id, mem_size):
self._cretate_frame_tree(acquired_bytes=mem_size)
self._create_frame_tree(acquired_bytes=mem_size)

def _cretate_frame_tree(self, used_bytes=0, acquired_bytes=0):
def _create_frame_tree(self, used_bytes=0, acquired_bytes=0):
self._root.used_bytes += used_bytes
self._root.acquired_bytes += acquired_bytes
parent = self._root
Expand Down

0 comments on commit 5f6b597

Please sign in to comment.