/generic_in_c

imitate the file structure of TH that under the pytorch

Primary LanguageC

generic_in_c

看TH文件夹下的文件,会发现 TH 下有一个 THStorage.cTHStorage.h 文件,generic 文件夹下又有两个同名的 THStorage.cTHStorage.h 文件。
其实外层的THStorage.c会包含内层的THStorage.c文件,然后结合THGenerateAllTypes.h文件,将内层的泛型函数 膨胀 成各种类型的函数。

下面是文件的对应关系:

generic.c TH\generic\THStorage.c
generic.h TH\generic\THStorage.h
show_types.c TH\THStorage.c
show_types.h TH\THStorage.h
all_types.h TH\THGenerateAllTypes.h
generic.c下的函数 **show_real(real number)** 会根据 real 的类型自动替换成show_int(int number) 或 show_double(double number)。

使用 gcc main.c show_types.c -o demo
然后运行 demo 可成功运行。

写了篇博客 pytorch下的lib库 源码阅读笔记(1) 记录学习过程。