![](https://img-blog.csdnimg.cn/20201014180756724.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
源码阅读
文章平均质量分 93
Chris_zhangrx
读书郎
展开
-
[PyTorch 源码阅读] —— TensorIterator
介绍 TensorIterator原创 2023-03-04 17:55:47 · 1173 阅读 · 1 评论 -
[源码阅读] TensorRT - ONNX parser
文章目录ONNX parser相关内容:[源码阅读] TensorRT —— Caffe Parser、PluginONNX parserTensorRT 的 onnx parser 是一个独立的开源项目。 onnx-tensorrt:https://github.com/onnx/onnx-tensorrt这里映射关系还是首先建立起来一个个映射函数,然后通过 op_type 来找到对应的映射函数。typedef ValueOrStatus<std::vector<TensorOrW原创 2022-03-07 08:58:38 · 826 阅读 · 0 评论 -
[源码阅读] TensorRT —— Caffe Parser、Plugin
文章目录前言Caffe parserPlugin本文主要从代码层面对 TensorRT 的源码进行学习,试图从中梳理出一点实现思路以及实现细节吧。个人水平有限,主要是从这个过程中学习为主,若有理解不对的地方欢迎交流指正。注:本文并不涉及到具体功能性的介绍,例如如何一步步去添加 Plugin , 或者一些具体的接口要如何使用等。前言TensorRT 源码部分主要开源出来了 Parser 部分以及 Plugin ,并且给出了相关的 demo 和一系列的 sample 供使用者可以快速学习。推理引擎其实原创 2022-03-05 16:31:10 · 1114 阅读 · 0 评论 -
Caffe 源码(一)—— caffe.proto 注解
参考内容:https://blog.csdn.net/mac_lzq/article/details/63725009使用以下语句可以对 caffe.proto 文件进行编译, 生成 caffe.pb.h 和 caffe.pb.cc 两个文件。 protoc -I=. --cpp_out=. ./caffe.proto以下内容是对 Caffe 框架下 src/caffe/proto 目...原创 2018-07-10 22:03:49 · 783 阅读 · 0 评论 -
Caffe源码(三)—— Blob
Caffe 源码 —— blob.hpp/cppsyncedmem.hpp/cpp在介绍 Blob 之前得先说一说 syncedmem.hpp/cpp,syncedmem 文件中定义了用于数据 CPU 和 GPU 之间的数据同步的 SyncedMemory 类,而这也是后面网络参数,梯度数据传输的基础。主要包含了:CaffeMallocHost 内存申请函数CaffeFreeHost...原创 2019-11-22 16:24:11 · 277 阅读 · 0 评论 -
Caffe 源码(二) —— common 文件
caffe 源码 —— common.hpp/cpp这里主要是对 caffe 框架源码进行梳理与学习(主要是 CPU 模式下的,所以暂时还不涉及 CUDA,cudnn 编程),不同时期回看源码收获不同,水平有限,如有错误之处还请留言指正交流。首先的切入点就是 common.hpp, 其包含在 blob.hpp 和 layer.cpp 头文件中,而 blob.hpp 和 layer.cpp 这两...原创 2019-11-14 19:56:08 · 444 阅读 · 0 评论 -
[Pytorch 源码阅读] ——Pytorch python 接口的 C 扩展
前言在前面的博客中介绍了 Tensor 相关类的 C++ 实现,本文主要介绍 Pytorch 的 Tensor 是如何在 Python 中使用起来的,更大一点的题目就是 Pytorch 中 python 的 C 扩展,说明 Pytorch 是如何通过 python 接口来调用相关 C++ 实现的。Python c 扩展在开展正式介绍之前,需要先了解一些前提知识,就是标准的 Python c 扩展的一些内容。python 中模块(module)和包(package)的简介:模块(module)就是原创 2021-07-26 10:19:26 · 1497 阅读 · 1 评论 -
[Pytorch 源码阅读] —— TH中的 c 语言泛型编程
前言基于 pytorch 1.10.0 版本,mastercommit 号:047925dac1c07a0ad2c86c281fac5610b084d1bd万事开头难,还是咬着牙开始了 Pytorch 的源码阅读内容,虽然感觉难度很大,而且有点无从下手,希望坚持下去能有所进步!这里光源码编译就花了一些时间,尝试了 macbook,windows,和 linux 下 git clone 源码然后按照官方说明操作,但是基本碍于系统或者网络问题,git 拉第三方以来的时候会有很多问题,最后只是在 li原创 2021-06-20 22:12:01 · 516 阅读 · 1 评论 -
[Pytorch 源码阅读] —— Tensor C++相关实现
文章目录版本介绍概念介绍源码分析intrusive_ptr 类TensorImpl 类StorageImpl 类总结版本介绍因为 Pytorch 版本迭代还是蛮快的,这里给出我源码阅读的版本及 commit 号以供参考。master 分支,commit 号: 047925dac1c07a0ad2c86c281fac5610b084d1bd概念介绍Tensor 是 Pytorch 核心的数据结构,使用过 pytorch 的同学想必都不会陌生,它可以包含 scalar type 的数据(例如 floa原创 2021-07-25 21:44:08 · 3790 阅读 · 3 评论 -
[Pytorch 源码阅读] ——torch.trace.jit 接口 C++ 源码阅读
文章目录前言torch::jit::TypeClassTypec10::ivalue::Objecttorch::jit::Objecttorch::jit::Module转化过程总结前言本文主要是从 torch.jit.trace 接口,打开 Pytorch JIT 的大门,介绍在正常 nn.module 经过 Trace 之后形成 scriptModule 的过程和涉及到的 C++ 类,因为涉及到的内容蛮多的,所以这里就从源码的角度介绍了一些相对比较重要,或者是可以帮助我们理解的类。在开始学习之前原创 2021-08-23 08:26:21 · 3406 阅读 · 0 评论 -
[Pytorch 源码阅读] —— 谈谈 dispatcher(二)
文章目录前言源码阅读IValue 类schemaOperatorHandleKernelFunctionOperatorEntryDispatcher内部调用逻辑kernel 是如何注册上的根据数据类型再次分发总结前言这里接上文 [Pytorch 源码阅读] —— 谈谈 dispatcher 中对 Pytorch 中 dispatcher 的相关概念介绍,主要是对相关概念和类间的关系做一个概述,没有读过前面文章的同学建议先读一下前面的文章。这里就是深入各个类的源码,对这些类做一个详细的解读。文章篇幅会原创 2021-08-08 11:40:35 · 2358 阅读 · 1 评论 -
[Pytorch 源码阅读] —— 谈谈 dispatcher(一)
文章目录概念介绍diapatch key 的表示和计算dispatch table 注册boxing 和 unboxing源码分析这篇文章的内容主要还是基于 EdWard z. yang 的 Let’s talk about the PyTorch dispatcher 来梳理一下 Pytorch dispatcher 相关的内容学习以及源码阅读。概念介绍dispatcher 可以理解为分发器,可以根据关于 tensor 输入的一些信息来决定要调用哪一块的程序。其主要是通过分发表(dispatch原创 2021-08-07 19:06:06 · 2224 阅读 · 0 评论