Memory management

Torch_npu提供内存管理相关的部分接口，具体可参考torch.cuda。