提交 21ccb530 authored 作者: Pascal Lamblin's avatar Pascal Lamblin

Add ignored "allow_downcast" parameter to cuda shared constructors.

This should unbreak GPU shared variables, broken after [c4a975e14f7c].
上级 a31a572e
......@@ -82,7 +82,8 @@ class CudaNdarraySharedVariable(SharedVariable, _operators):
CudaNdarrayType.SharedVariable = CudaNdarraySharedVariable
def cuda_shared_constructor(value, name=None, strict=False, borrow=False, broadcastable=None):
def cuda_shared_constructor(value, name=None, strict=False,
allow_downcast=None, borrow=False, broadcastable=None):
"""SharedVariable Constructor for TensorType"""
# THIS CONSTRUCTOR TRIES TO CAST VALUE TO A FLOAT32, WHICH THEN GOES ONTO THE CARD
......@@ -112,7 +113,8 @@ def cuda_shared_constructor(value, name=None, strict=False, borrow=False, broadc
raise
return rval
def float32_shared_constructor(value, name=None, strict=False, borrow=False, broadcastable=None):
def float32_shared_constructor(value, name=None, strict=False,
allow_downcast=None, borrow=False, broadcastable=None):
"""SharedVariable Constructor for TensorType"""
# if value isn't a float32 ndarray, then raise
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论