Swig简单实验和理解

最新推荐文章于 2023-03-10 20:55:19 发布

天使在飞-666

最新推荐文章于 2023-03-10 20:55:19 发布

阅读量1.1k

点赞数

分类专栏： Tensorflow专题文章标签： tensorflow python 深度学习 c++

本文链接：https://blog.csdn.net/weixin_37614967/article/details/122266053

版权

Tensorflow专题专栏收录该内容

1 篇文章 0 订阅

订阅专栏

一问题起源

在看tensorflow的源码的时候，发现python调用c++框架是如何实现的引起了我的好奇，发现编译的时候使用的是Swig

操作环境

为了搞清楚swig使用方法，简单做了一个实验验证Swig; Swig是一个独立库可以让用户使用python调用c++程序；

环境问题

tensorflow本身在编译依赖的时候都是直接下载对应的库放在一个临时编译目录

比如我的编译路径如下，注意红色的部分每个工程编译都是一个独立的目录，其实tensorflow这样搞也有好处，多个工程之间不耦合；但是对我们验证Swig程序会有影响，因为不在标准的lib库下

/root/.cache/bazel/_bazel_root/014e8ed0b29435778e37e53e83f1c80c/execroot/org_tensorflow/bazel-out/k8-opt/bin/tensorflow/python/tools/inspect_checkpoint.runfiles/org_tensorflow/tensorflow/python/pywrap_tensorflow_internal.py

解决办法：

只是做测试，所以干脆暴力单独拷贝swig库测试

创建独立测试目录 /home/test_swig；拷贝可执行程序swig到上面的/home/test_swig目录，执行过程中可能会报错，直接去tensorflow路径下找对应的文件就可以了，肯定能找到的，找到后拷贝到一起的样子如下：

二编写测试程序

1. 比如我们写一个example.c的文件

/* File : example.c */

double  My_variable  = 3.0;

/* Compute factorial of n */
int fact(int n) {
  if (n <= 1)
    return 1;
  else
    return n*fact(n-1);
}

/* Compute n mod m */
int my_mod(int n, int m) {
  return(n % m);
}

2. 导出文件写一个example.i

/* File : example.i */
%module example
%{
/* Put headers and other declarations here */
extern double My_variable;
extern int    fact(int);
extern int    my_mod(int n, int m);
%}
 
extern double My_variable;
extern int    fact(int);
extern int    my_mod(int n, int m);

3. 编译执行

$swig -python example.i
$gcc -c -fpic example.c example_wrap.c -I/usr/local/python36/include/python3.6m/
$gcc -shared example.o example_wrap.o -o _example.so
$python    
>>> import example
>>> example.fact(4)

最后生成出的文件

so _example.so

example_wrap.c

example.py