Skip to content
项目
群组
代码片段
帮助
当前项目
正在载入...
登录 / 注册
切换导航面板
P
pytensor
项目
项目
详情
活动
周期分析
仓库
仓库
文件
提交
分支
标签
贡献者
图表
比较
统计图
议题
0
议题
0
列表
看板
标记
里程碑
合并请求
0
合并请求
0
CI / CD
CI / CD
流水线
作业
日程
统计图
Wiki
Wiki
代码片段
代码片段
成员
成员
折叠边栏
关闭边栏
活动
图像
聊天
创建新问题
作业
提交
问题看板
Open sidebar
testgroup
pytensor
Commits
b1cc9fd8
提交
b1cc9fd8
authored
8月 13, 2014
作者:
Arnaud Bergeron
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
Move the _apply versions of the function under CLinkerOp
and clarify/update documentation.
上级
db7bffb4
隐藏空白字符变更
内嵌
并排
正在显示
3 个修改的文件
包含
85 行增加
和
72 行删除
+85
-72
cc.py
theano/gof/cc.py
+0
-4
cmodule.py
theano/gof/cmodule.py
+1
-3
op.py
theano/gof/op.py
+84
-65
没有找到文件。
theano/gof/cc.py
浏览文件 @
b1cc9fd8
...
...
@@ -535,7 +535,6 @@ class CLinker(link.Linker):
for
variable
in
self
.
variables
:
# it might be possible to inline constant variables as C literals
## if getattr(variable, 'constant', False):
# policy = [[what to declare in the struct,
# what to do at construction,
# what to do at destruction],
...
...
@@ -545,9 +544,6 @@ class CLinker(link.Linker):
if
variable
in
self
.
inputs
:
# we need to extract the new inputs at each run
# they do not need to be relayed to Python, so we don't sync
# if isinstance(variable, Constant):
# raise TypeError("Inputs to CLinker cannot be Constant.",
# variable)
policy
=
[[
get_nothing
,
get_nothing
,
get_nothing
],
[
get_c_declare
,
get_c_extract
,
get_c_cleanup
]]
elif
variable
in
self
.
orphans
:
...
...
theano/gof/cmodule.py
浏览文件 @
b1cc9fd8
...
...
@@ -1522,8 +1522,6 @@ def gcc_llvm():
It don't support all g++ parameters even if it support many of them.
"""
if
gcc_llvm
.
is_llvm
is
None
:
pass
p
=
None
try
:
p_out
=
output_subprocess_Popen
([
'g++'
,
'--version'
])
output
=
p_out
[
0
]
+
p_out
[
1
]
...
...
@@ -1535,9 +1533,9 @@ def gcc_llvm():
# compile when g++ is not available. If this happen, it
# will crash later so supposing it is not llvm is "safe".
output
=
b
(
''
)
del
p
gcc_llvm
.
is_llvm
=
b
(
"llvm"
)
in
output
return
gcc_llvm
.
is_llvm
gcc_llvm
.
is_llvm
=
None
...
...
theano/gof/op.py
浏览文件 @
b1cc9fd8
...
...
@@ -126,22 +126,6 @@ class CLinkerObject(object):
"""
return
()
def
c_code_cache_version_apply
(
self
,
node
):
"""Return a tuple of integers indicating the version of this Op.
An empty tuple indicates an 'unversioned' Op that will not be cached between processes.
The cache mechanism may erase cached modules that have been superceded by newer
versions. See `ModuleCache` for details.
:note: See also `c_code_cache_version()`
:note: This function overrides `c_code_cache_version` unless it explicitly calls
`c_code_cache_version`. The default implementation simply calls `c_code_cache_version`
and ignores the `node` argument.
"""
return
self
.
c_code_cache_version
()
def
c_compile_args
(
self
):
"""Optional: Return a list of compile args recommended to compile the
code returned by other methods in this class.
...
...
@@ -187,18 +171,6 @@ class CLinkerObject(object):
self
.
__class__
.
__name__
)
def
c_init_code_apply
(
self
,
node
,
name
):
"""
Optional: return a list of code snippets specific to the apply
to be inserted in module initialization.
:Exceptions:
- `MethodNotDefined`: the subclass does not override this method
"""
raise
utils
.
MethodNotDefined
(
"c_init_code_apply"
,
type
(
self
),
self
.
__class__
.
__name__
)
class
CLinkerOp
(
CLinkerObject
):
"""
Interface definition for `Op` subclasses compiled by `CLinker`.
...
...
@@ -217,24 +189,25 @@ class CLinkerOp(CLinkerObject):
:Parameters:
`node` : Apply instance
WRITEME
`name` : WRITEME
WRITEME
The node for which we are compiling the current c_code.
The same Op may be used in more than one node.
`name` : A string
A name that is automatically assigned and guaranteed to be unique.
`inputs` : list of strings
There is a string for each input of the function, and the string is the name of a C
`PyObject` variable pointing to that input.
There is a string for each input of the function, and the
string is the name of a C `PyObject` variable pointing to
that input.
`outputs` : list of strings
Each string is the name of a `PyObject` pointer where the Op should
store its variables. As of version 0.4.0, this pointer could be
NULL, or contain an object allocated during a previous call to the
same function, unchanged from the end of the previous execution.
In a future version, there will be no guarantee on where that
object will be created (it could be allocated during a previous
execution, or by another Op, by the Mode, etc.). It will still
be of an appropriate Type (in the Theano sense) to store the output
of the computation: for instance, for a TensorVariable, it will be a
Numpy ndarray with the right number of dimensions, and the right dtype.
However, its shape, or stride pattern, could not be adequate.
Each string is the name of a `PyObject` pointer where the
Op should store its variables. This pointer may either be
NULL, indicating that the Op must allocate appropriate
objects or it may point to preallocated objects of the
right type and number of dimensions. In the case of
preallocated objects, the Op must make sure that the shape
and strides meet requirements, and in the case they don't
either reallocate the object inplace or free it and
allocate an appropriate output. The type and number of
dimensions are guaranteed to be appropriate.
`sub` : dict of strings
extra symbols defined in `CLinker` sub symbols (such as 'fail').
WRITEME
...
...
@@ -246,10 +219,28 @@ class CLinkerOp(CLinkerObject):
raise
utils
.
MethodNotDefined
(
'
%
s.c_code'
\
%
self
.
__class__
.
__name__
)
def
c_code_cleanup
(
self
,
node
,
name
,
inputs
,
outputs
,
sub
):
"""Optional: Return C code to run after c_code, whether it failed or not.
def
c_code_cache_version_apply
(
self
,
node
):
"""Return a tuple of integers indicating the version of this Op.
An empty tuple indicates an 'unversioned' Op that will not be
cached between processes.
The cache mechanism may erase cached modules that have been
superceded by newer versions. See `ModuleCache` for details.
:note: See also `c_code_cache_version()`
:note: This function overrides `c_code_cache_version` unless
it explicitly calls `c_code_cache_version`. The
default implementation simply calls
`c_code_cache_version` and ignores the `node` argument.
"""
return
self
.
c_code_cache_version
()
QUESTION: is this function optional?
def
c_code_cleanup
(
self
,
node
,
name
,
inputs
,
outputs
,
sub
):
"""
Optional: Return C code to run after c_code, whether it failed
or not.
This is a convenient place to clean up things allocated by c_code().
...
...
@@ -259,39 +250,45 @@ class CLinkerOp(CLinkerObject):
`name` : WRITEME
WRITEME
`inputs` : list of strings
There is a string for each input of the function, and the string is the name of a C
`PyObject` variable pointing to that input.
There is a string for each input of the function, and the
string is the name of a C `PyObject` variable pointing to
that input.
`outputs` : list of strings
Each string is the name of a `PyObject` pointer where the Op should store its
variables. This pointer could be NULL, or contain an object of the right
Type (in the Theano sense) to store the output of the computation.
For instance, for a TensorVariable, it will be a Numpy ndarray with
the right number of dimensions, and the right dtype. However, its
shape, or stride pattern, could not be adequate.
It could be unchanged from the end of the previous execution, or allocated
by another Op, or by the Mode.
Each string is the name of a `PyObject` pointer where the
Op should store its variables. This pointer could be NULL,
or contain an object of the right Type (in the Theano
sense) to store the output of the computation. For
instance, for a TensorVariable, it will be a Numpy ndarray
with the right number of dimensions, and the right
dtype. However, its shape, or stride pattern, could not be
adequate. It could be unchanged from the end of the
previous execution, or allocated by another Op, or by the
Mode.
`sub` : dict of strings
extra symbols defined in `CLinker` sub symbols (such as 'fail').
WRITEME
WRITEME
:Exceptions:
- `MethodNotDefined`: the subclass does not override this method
"""
raise
utils
.
MethodNotDefined
(
'
%
s.c_code_cleanup'
\
%
self
.
__class__
.
__name__
)
def
c_support_code_apply
(
self
,
node
,
name
):
"""Optional: Return utility code for use by an `Op` that will be inserted at global
scope, that can be specialized for the support of a particular `Apply` node.
"""Optional: Return utility code for use by an `Op` that will be
inserted at struct scope, that can be specialized for the
support of a particular `Apply` node.
:param node: an Apply instance in the graph being compiled
:param node_id: a string or number that serves to uniquely identify this node.
Symbol names defined by this support code should include the node_id, so that they can
be called from the c_code, and so that they do not cause name collisions.
:param name: a string or number that serves to uniquely
identify this node. Symbol names defined by this
support code should include the name, so that
they can be called from the c_code, and so that
they do not cause name collisions.
:note: This function is called in addition to c_support_code
and will supplement whatever is returned from there.
:Exceptions:
- `MethodNotDefined`: Subclass does not implement this method
...
...
@@ -300,6 +297,28 @@ class CLinkerOp(CLinkerObject):
raise
utils
.
MethodNotDefined
(
"c_support_code_apply"
,
type
(
self
),
self
.
__class__
.
__name__
)
def
c_init_code_apply
(
self
,
node
,
name
):
"""
Optional: return a code string specific to the apply
to be inserted in the struct initialization code.
:param node: an Apply instance in the graph being compiled
:param name: a string or number that serves to uniquely
identify this node. Symbol names defined by this
support code should include the name, so that
they can be called from the c_code, and so that
they do not cause name collisions.
:note: This function is called in addition to c_init_code
and will supplement whatever is returned from there.
:Exceptions:
- `MethodNotDefined`: the subclass does not override this method
"""
raise
utils
.
MethodNotDefined
(
"c_init_code_apply"
,
type
(
self
),
self
.
__class__
.
__name__
)
class
PureOp
(
object
):
"""
...
...
编写
预览
Markdown
格式
0%
重试
或
添加新文件
添加附件
取消
您添加了
0
人
到此讨论。请谨慎行事。
请先完成此评论的编辑!
取消
请
注册
或者
登录
后发表评论