src/ParamMap.cpp · xiongying/Halide

代码拉取完成,页面将自动刷新

扫描微信二维码支付

取消

支付完成

richgiteeai

Watch

不关注关注所有动态仅关注版本发行动态关注但不提醒动态

1 Star 0 Fork 3

xiongying/Halide

forked from Gitee 极速下载/Halide

代码 Wiki 统计流水线

服务

加入 Gitee

与超过 1400万开发者一起发现、参与优秀开源项目,私有仓库也完全免费 :)

免费加入

已有帐号? 立即登录

文件

main

分支 (1093)

标签 (17)

管理

main

xtensa-codegen

vksnk/dma-limit-channels

rootjalex/trs-codegen-cross

abadams/fix_7374

abadams/remove_hack_from_gpu_only_aottest

srj/gpu-cache

srj/generator_aot_gpu_multi_context_threaded

srj/xtensa-merge

abadams/vector_scan

abadams/fix_7365

darya-ver/ir-viz

vulkan-phase2-runtime

srj/param-map-deprecation

srj/rt-return-types

srj/main-vs2022

release/15.x

srj/param-map

abadams/ir_builder_unique_ptr

vksnk/restrict

v14.0.0

v13.0.4

v13.0.3

v13.0.2

v13.0.1

v13.0.0

v12.0.1

v12.0.0

v11.0.1

v11.0.0

v10.0.1

v10.0.0

release_2019_08_27

release_8.0.0

v8.0.0

release_2018_02_15

release_2013_11_11

克隆/下载

HTTPS SSH SVN SVN+SSH 下载ZIP

提示

下载代码请复制以下命令到终端执行

为确保你提交的代码身份被 Gitee 正确识别,请执行以下命令完成配置

git config --global user.name userName 
git config --global user.email userEmail

初次使用 SSH 协议进行代码克隆、推送等操作时,需按下述提示完成 SSH 配置

1 生成 RSA 密钥

2 获取 RSA 公钥内容,并配置到 SSH公钥中

在 Gitee 上使用 SVN,请访问使用指南

使用 HTTPS 协议时,命令行会出现如下账号密码验证步骤。基于安全考虑,Gitee 建议配置并使用私人令牌替代登录密码进行克隆、推送等操作

Username for 'https://gitee.com': userName

Password for 'https://userName@gitee.com': # 私人令牌

分支 1093

标签 17

Halide

src

ParamMap.cpp

ParamMap.cpp 1.55 KB

#include "ParamMap.h"
#include "ImageParam.h"

namespace Halide {

ParamMap::ParamMap(const std::initializer_list<ParamMapping> &init) {
 for (const auto &pm : init) {
 if (pm.parameter != nullptr) {
 mapping[*pm.parameter] = ParamArg(pm);
 } else if (pm.buf_out_param == nullptr) {
 // TODO: there has to be a way to do this without the const_cast.
 set(*pm.image_param, *const_cast<Buffer<> *>(&pm.buf), nullptr);
 } else {
 Buffer<> temp_undefined;
 set(*pm.image_param, temp_undefined, pm.buf_out_param);
 }
 }
}

void ParamMap::set(const ImageParam &p, const Buffer<> &buf, Buffer<> *buf_out_param) {
 Internal::Parameter v(p.type(), true, p.dimensions(), p.name());
 v.set_buffer(buf);
 ParamArg pa;
 pa.mapped_param = v;
 pa.buf_out_param = buf_out_param;
 mapping[p.parameter()] = pa;
};

const Internal::Parameter &ParamMap::map(const Internal::Parameter &p, Buffer<> *&buf_out_param) const {
 auto iter = mapping.find(p);
 if (iter != mapping.end()) {
 buf_out_param = iter->second.buf_out_param;
 return iter->second.mapped_param;
 } else {
 buf_out_param = nullptr;
 return p;
 }
}

Internal::Parameter &ParamMap::map(Internal::Parameter &p, Buffer<> *&buf_out_param) const {
 auto iter = mapping.find(p);
 if (iter != mapping.end()) {
 buf_out_param = iter->second.buf_out_param;
 return iter->second.mapped_param;
 } else {
 buf_out_param = nullptr;
 return p;
 }
}

} // namespace Halide

一键复制编辑原始数据按行查看历史

Steven Johnson 提交于 2020年04月07日 02:35 +08:00 . Tweak to ParamMap API

1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51

#include "ParamMap.h"
#include "ImageParam.h"

namespace Halide {

ParamMap::ParamMap(const std::initializer_list<ParamMapping> &init) {
 for (const auto &pm : init) {
 if (pm.parameter != nullptr) {
 mapping[*pm.parameter] = ParamArg(pm);
 } else if (pm.buf_out_param == nullptr) {
 // TODO: there has to be a way to do this without the const_cast.
 set(*pm.image_param, *const_cast<Buffer<> *>(&pm.buf), nullptr);
 } else {
 Buffer<> temp_undefined;
 set(*pm.image_param, temp_undefined, pm.buf_out_param);
 }
 }
}

void ParamMap::set(const ImageParam &p, const Buffer<> &buf, Buffer<> *buf_out_param) {
 Internal::Parameter v(p.type(), true, p.dimensions(), p.name());
 v.set_buffer(buf);
 ParamArg pa;
 pa.mapped_param = v;
 pa.buf_out_param = buf_out_param;
 mapping[p.parameter()] = pa;
};

const Internal::Parameter &ParamMap::map(const Internal::Parameter &p, Buffer<> *&buf_out_param) const {
 auto iter = mapping.find(p);
 if (iter != mapping.end()) {
 buf_out_param = iter->second.buf_out_param;
 return iter->second.mapped_param;
 } else {
 buf_out_param = nullptr;
 return p;
 }
}

Internal::Parameter &ParamMap::map(Internal::Parameter &p, Buffer<> *&buf_out_param) const {
 auto iter = mapping.find(p);
 if (iter != mapping.end()) {
 buf_out_param = iter->second.buf_out_param;
 return iter->second.mapped_param;
 } else {
 buf_out_param = nullptr;
 return p;
 }
}

} // namespace Halide