2014-09-25 145 views
2

sympy庫有這種很酷的方法,它將符號表達式轉換爲C代碼,編譯它,然後輸出調用該C方法的lambda函數。sympy中的參數限制ulfuncify

我做了這個簡單的演示,demo.py

import sympy 
from sympy.utilities.autowrap import ufuncify 
import sys 

N = int(sys.argv[1]) 
theta = [] 
values = [] 
for n in range(N): 
    theta.append(sympy.symbols('x%s' % n)) 
    values.append(n) 

summation = sum(theta) 
f = ufuncify(theta, summation) 
print f(*values)[0] 

它適用於小N

 
$ python demo.py 21 
210.0 

但對於較大N,我得到以下錯誤:

 
$ python demo.py 22 
Traceback (most recent call last): 
    File "demo.py", line 13, in 
    f = ufuncify(theta, summation) 
    File "/opt/user/x86_64/Python-2.7.3/lib/python2.7/site-packages/sympy/utilities/autowrap.py", line 485, in ufuncify 
    return autowrap(C.Equality(y[i], f(*args)), **kwargs) 
    File "/opt/user/x86_64/Python-2.7.3/lib/python2.7/site-packages/sympy/utilities/autowrap.py", line 403, in autowrap 
    return code_wrapper.wrap_code(routine, helpers=helps) 
    File "/opt/user/x86_64/Python-2.7.3/lib/python2.7/site-packages/sympy/utilities/autowrap.py", line 139, in wrap_code 
    self._process_files(routine) 
    File "/opt/user/x86_64/Python-2.7.3/lib/python2.7/site-packages/sympy/utilities/autowrap.py", line 158, in _process_files 
    " ".join(command), e.output)) 
sympy.utilities.autowrap.CodeWrapError: Error while executing command: f2py -m wrapper_module_0 -c wrapped_code_0.f90. Command output is: 
running build 
running config_cc 
unifing config_cc, config, build_clib, build_ext, build commands --compiler options 
running config_fc 
unifing config_fc, config, build_clib, build_ext, build commands --fcompiler options 
running build_src 
build_src 
building extension "wrapper_module_0" sources 
f2py options: [] 
f2py:> /tmp/tmpKbJQuO/src.linux-x86_64-2.7/wrapper_module_0module.c 
creating /tmp/tmpKbJQuO 
creating /tmp/tmpKbJQuO/src.linux-x86_64-2.7 
Reading fortran codes... 
     Reading file 'wrapped_code_0.f90' (format:free) 
Post-processing... 
     Block: wrapper_module_0 
         Block: autofunc 
Post-processing (stage 2)... 
Building modules... 
     Building module "wrapper_module_0"... 
       Constructing wrapper function "autofunc"... 
        y_15 = autofunc(x_16,x1,x10,x11,x12,x13,x14,x15,x16,x17,x18,x19,x2,x20,x21,x3,x4,x5,x6,x7,x8,x9,[m_17]) 
     Wrote C/API module "wrapper_module_0" to file "/tmp/tmpKbJQuO/src.linux-x86_64-2.7/wrapper_module_0module.c" 
    adding '/tmp/tmpKbJQuO/src.linux-x86_64-2.7/fortranobject.c' to sources. 
    adding '/tmp/tmpKbJQuO/src.linux-x86_64-2.7' to include_dirs. 
copying /opt/user/x86_64/Python-2.7.3/lib/python2.7/site-packages/numpy/f2py/src/fortranobject.c -> /tmp/tmpKbJQuO/src.linux-x86_64-2.7 
copying /opt/user/x86_64/Python-2.7.3/lib/python2.7/site-packages/numpy/f2py/src/fortranobject.h -> /tmp/tmpKbJQuO/src.linux-x86_64-2.7 
build_src: building npy-pkg config files 
running build_ext 
customize UnixCCompiler 
customize UnixCCompiler using build_ext 
customize Gnu95FCompiler 
Found executable /opt/user/x86_64/gcc-4.7.2/bin/gfortran 
customize Gnu95FCompiler 
customize Gnu95FCompiler using build_ext 
building 'wrapper_module_0' extension 
compiling C sources 
C compiler: gcc -pthread -fno-strict-aliasing -g -O2 -DNDEBUG -g -fwrapv -O3 -Wall -Wstrict-prototypes -fPIC 

creating /tmp/tmpKbJQuO/tmp 
creating /tmp/tmpKbJQuO/tmp/tmpKbJQuO 
creating /tmp/tmpKbJQuO/tmp/tmpKbJQuO/src.linux-x86_64-2.7 
compile options: '-I/tmp/tmpKbJQuO/src.linux-x86_64-2.7 -I/opt/user/x86_64/Python-2.7.3/lib/python2.7/site-packages/numpy/core/include -I/opt/user/x86_64/Python-2.7.3/include/python2.7 -c' 
gcc: /tmp/tmpKbJQuO/src.linux-x86_64-2.7/wrapper_module_0module.c 
In file included from /opt/user/x86_64/Python-2.7.3/lib/python2.7/site-packages/numpy/core/include/numpy/ndarraytypes.h:1728:0, 
       from /opt/user/x86_64/Python-2.7.3/lib/python2.7/site-packages/numpy/core/include/numpy/ndarrayobject.h:17, 
       from /opt/user/x86_64/Python-2.7.3/lib/python2.7/site-packages/numpy/core/include/numpy/arrayobject.h:15, 
       from /tmp/tmpKbJQuO/src.linux-x86_64-2.7/fortranobject.h:13, 
       from /tmp/tmpKbJQuO/src.linux-x86_64-2.7/wrapper_module_0module.c:18: 
/opt/user/x86_64/Python-2.7.3/lib/python2.7/site-packages/numpy/core/include/numpy/npy_deprecated_api.h:11:2: warning: #warning "Using deprecated NumPy API, disable it by #defining NPY_NO_DEPRECATED_API NPY_1_7_API_VERSION" [-Wcpp] 
/tmp/tmpKbJQuO/src.linux-x86_64-2.7/wrapper_module_0module.c:111:12: warning: âpy_sizeâefined but not used [-Wunused-function] 
gcc: /tmp/tmpKbJQuO/src.linux-x86_64-2.7/fortranobject.c 
In file included from /opt/user/x86_64/Python-2.7.3/lib/python2.7/site-packages/numpy/core/include/numpy/ndarraytypes.h:1728:0, 
       from /opt/user/x86_64/Python-2.7.3/lib/python2.7/site-packages/numpy/core/include/numpy/ndarrayobject.h:17, 
       from /opt/user/x86_64/Python-2.7.3/lib/python2.7/site-packages/numpy/core/include/numpy/arrayobject.h:15, 
       from /tmp/tmpKbJQuO/src.linux-x86_64-2.7/fortranobject.h:13, 
       from /tmp/tmpKbJQuO/src.linux-x86_64-2.7/fortranobject.c:2: 
/opt/user/x86_64/Python-2.7.3/lib/python2.7/site-packages/numpy/core/include/numpy/npy_deprecated_api.h:11:2: warning: #warning "Using deprecated NumPy API, disable it by #defining NPY_NO_DEPRECATED_API NPY_1_7_API_VERSION" [-Wcpp] 
compiling Fortran sources 
Fortran f77 compiler: /opt/user/x86_64/gcc-4.7.2/bin/gfortran -Wall -ffixed-form -fno-second-underscore -fPIC -O3 -funroll-loops 
Fortran f90 compiler: /opt/user/x86_64/gcc-4.7.2/bin/gfortran -Wall -fno-second-underscore -fPIC -O3 -funroll-loops 
Fortran fix compiler: /opt/user/x86_64/gcc-4.7.2/bin/gfortran -Wall -ffixed-form -fno-second-underscore -Wall -fno-second-underscore -fPIC -O3 -funroll-loops 
compile options: '-I/tmp/tmpKbJQuO/src.linux-x86_64-2.7 -I/opt/user/x86_64/Python-2.7.3/lib/python2.7/site-packages/numpy/core/include -I/opt/user/x86_64/Python-2.7.3/include/python2.7 -c' 
gfortran:f90: wrapped_code_0.f90 
wrapped_code_0.f90:1.133: 

4, x15, x16, x17, x18, x19, x2, x20, x21, x3, x4, x5, x6, x7, x8, x9, y_15 
                      1 
Warning: Line truncated at (1) 
wrapped_code_0.f90:1.132: 

14, x15, x16, x17, x18, x19, x2, x20, x21, x3, x4, x5, x6, x7, x8, x9, y_15 
                      1 
Error: Unexpected junk in formal argument list at (1) 
wrapped_code_0.f90:33.3: 

end subroutine 
    1 
Error: Expecting END PROGRAM statement at (1) 
Error: Unexpected end of file in 'wrapped_code_0.f90' 
wrapped_code_0.f90:1.133: 

4, x15, x16, x17, x18, x19, x2, x20, x21, x3, x4, x5, x6, x7, x8, x9, y_15 
                      1 
Warning: Line truncated at (1) 
wrapped_code_0.f90:1.132: 

14, x15, x16, x17, x18, x19, x2, x20, x21, x3, x4, x5, x6, x7, x8, x9, y_15 
                      1 
Error: Unexpected junk in formal argument list at (1) 
wrapped_code_0.f90:33.3: 

end subroutine 
    1 
Error: Expecting END PROGRAM statement at (1) 
Error: Unexpected end of file in 'wrapped_code_0.f90' 
error: Command "/opt/user/x86_64/gcc-4.7.2/bin/gfortran -Wall -fno-second-underscore -fPIC -O3 -funroll-loops -I/tmp/tmpKbJQuO/src.linux-x86_64-2.7 -I/opt/user/x86_64/Python-2.7.3/lib/python2.7/site-packages/numpy/core/include -I/opt/user/x86_64/Python-2.7.3/include/python2.7 -c -c wrapped_code_0.f90 -o /tmp/tmpKbJQuO/wrapped_code_0.o" failed with exit status 1 

我怎樣才能使這個工作?

+1

'警告:行被截斷(1)' - 聽起來像一個人只能處理這麼多的變量。我希望這是一個你應該對sympy提出的錯誤。 – mdurant 2014-09-25 18:01:44

+0

看起來像一個bug ---生成的fortran代碼的行太長。 ufuncify應該包裝這些,但顯然它沒有。 – 2014-09-25 21:27:33

回答

0

毫無疑問,您在Sympy中發現了一個錯誤,但是您可以通過將標誌傳遞給後端來解決它。根據http://docs.sympy.org/latest/modules/utilities/autowrap.html的文檔,ufuncify將關鍵字參數傳遞給autowrap。 autowrapflags關鍵字參數看起來很有希望。反過來,f2py(默認後端)具有可讓您將標誌轉發給編譯器的選項。

使用gfortran如編譯器,我想嘗試以下操作:

f = ufuncify(theta, summation, flags='--f90flags="--ffree-line-length-none"') 

(這是目前沒有測試。)

爲清楚:--f90flags是f2py的一個選項,可用於指定編譯器的選項。 --ffree-line-length-none標誌被轉發給編譯器,並將指示gfortran永不截斷源代碼中的行。