2 Star 0 Fork 0

huangzhenyu/npu_cache_allocator

加入 Gitee
与超过 1200万 开发者一起发现、参与优秀开源项目,私有仓库也完全免费 :)
免费加入
文件
该仓库未声明开源许可证文件(LICENSE),使用请关注具体项目描述及其代码上游依赖。
克隆/下载
Decorator.cpp 735 Bytes
一键复制 编辑 原始数据 按行查看 历史
huangzhenyu 提交于 2024-02-07 20:32 . add files
//
// Created by hzy on 2024/2/7.
//
#include "Decorator.h"
void Decorator::memory_recorder_start() {
MemoryRecorder::start_record();
}
void Decorator::memory_recorder_end() {
MemoryRecorder::end_record();
}
void Decorator::malloc_recorder_start() {
MallocRecorder::start_record();
}
void Decorator::malloc_recorder_end() {
MallocRecorder::end_record();
}
// 从precise_match_start 到 precise_match_end一直精准匹配,将precise_match_start到precise_match_end之间的tensor全部放入到long池子中
// 精准匹配开始的接口
void Decorator::precise_match_start() {
is_precise_match = true;
}
// 精准匹配结束的接口
void Decorator::precise_match_end() {
is_precise_match = false;
}
Loading...
马建仓 AI 助手
尝试更多
代码解读
代码找茬
代码优化
1
https://gitee.com/heziiop/npu_cache_allocator.git
[email protected]:heziiop/npu_cache_allocator.git
heziiop
npu_cache_allocator
npu_cache_allocator
master

搜索帮助