#912 add mem_get_info

Merged

zoulq merged 1 commits from erpim_0402 into master 8 months ago

hanjr reviewed 9 months ago

mindtorch/torch/tensor.py

Erpim changed title from ~~add mem_get_info~~ to [wip]add mem_get_info 9 months ago

Erpim reviewed 9 months ago

mindtorch/torch/cuda/__init__.py

Erpim changed title from ~~[wip]add mem_get_info~~ to add mem_get_info 8 months ago

\lgtm

zoulq referenced this issue from a commit 8 months ago

zoulq merged commit 9d0607ff38 into master 8 months ago

No reviewers

No Label

No Milestone

No Assignees

2 Participants

Notifications

Due Date

No due date set.

Dependencies

This pull request currently doesn't have any dependencies.

@@ -460,2 +460,3 @@
         other_ms = cast_to_ms_tensor(other)
         out = tensor_ms.__mul__(other_ms)
         if tensor_ms.dtype == mstype.bool_ and other_ms.dtype == mstype.bool_:

@@ -103,9 +105,10 @@ def is_initialized():
     backend = get_backend()
     return ms.hal.is_initialized(backend)
 @lru_cache(_GLOBAL_LRU_CACHE_SIZE_NN)