提交 1ae48129 authored 作者: nouiz's avatar nouiz

Merge pull request #779 from dwf/openmp_compile_conditional

Make #include of omp.h compile-time conditional.
...@@ -40,10 +40,14 @@ else: ...@@ -40,10 +40,14 @@ else:
default_openmp = count > 1 default_openmp = count > 1
AddConfigVar('openmp', AddConfigVar('openmp',
"Enable or not parallel computation on the CPU with OpenMP. " "Enable (or not) parallel computation on the CPU with OpenMP. "
"It is the default value used when creating an Op that support it" "This is the default value used when creating an Op that "
". The best is to define it via Theano configuration " "supports OpenMP parallelization. It is preferable to define it "
"file or with the environment variable THEANO_FLAGS.", "via the Theano configuration file ~/.theanorc or with the "
"environment variable THEANO_FLAGS. Parallelization is only "
"done for some operations that implement it, and even for "
"operations that implement parallelism, each operation is free "
"to respect this flag or not.",
BoolParam(default_openmp), BoolParam(default_openmp),
in_c_key=False, in_c_key=False,
) )
......
...@@ -883,7 +883,7 @@ class ConvOp(Op): ...@@ -883,7 +883,7 @@ class ConvOp(Op):
return [din, dw] return [din, dw]
def c_headers(self): def c_headers(self):
return ['<numpy/noprefix.h>', '<iostream>', '<sstream>', '<omp.h>' ] return ['<numpy/noprefix.h>', '<iostream>', '<sstream>']
def c_code_cache_version(self): def c_code_cache_version(self):
return (8, self.openmp) return (8, self.openmp)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论