提交 513f9fd1 authored 作者: Frederic Bastien's avatar Frederic Bastien

Remove warning by warn_float64 when we want the upcast to float64.

上级 dd060035
...@@ -50,6 +50,7 @@ class GpuGemv(BlasOp): ...@@ -50,6 +50,7 @@ class GpuGemv(BlasOp):
A = as_gpuarray_variable(A, ctx_name) A = as_gpuarray_variable(A, ctx_name)
x = as_gpuarray_variable(x, ctx_name) x = as_gpuarray_variable(x, ctx_name)
y = as_gpuarray_variable(y, ctx_name) y = as_gpuarray_variable(y, ctx_name)
with theano.configparser.change_flags(warn_float64='ignore'):
alpha = as_tensor_variable(alpha).astype('float64') alpha = as_tensor_variable(alpha).astype('float64')
beta = as_tensor_variable(beta).astype('float64') beta = as_tensor_variable(beta).astype('float64')
...@@ -162,6 +163,7 @@ class GpuGemm(BlasOp): ...@@ -162,6 +163,7 @@ class GpuGemm(BlasOp):
A = as_gpuarray_variable(A, ctx_name) A = as_gpuarray_variable(A, ctx_name)
B = as_gpuarray_variable(B, ctx_name) B = as_gpuarray_variable(B, ctx_name)
C = as_gpuarray_variable(C, ctx_name) C = as_gpuarray_variable(C, ctx_name)
with theano.configparser.change_flags(warn_float64='ignore'):
alpha = as_tensor_variable(alpha).astype('float64') alpha = as_tensor_variable(alpha).astype('float64')
beta = as_tensor_variable(beta).astype('float64') beta = as_tensor_variable(beta).astype('float64')
assert alpha.ndim == 0 assert alpha.ndim == 0
...@@ -242,6 +244,7 @@ class GpuGer(BlasOp): ...@@ -242,6 +244,7 @@ class GpuGer(BlasOp):
A = as_gpuarray_variable(A, ctx_name) A = as_gpuarray_variable(A, ctx_name)
x = as_gpuarray_variable(x, ctx_name) x = as_gpuarray_variable(x, ctx_name)
y = as_gpuarray_variable(y, ctx_name) y = as_gpuarray_variable(y, ctx_name)
with theano.configparser.change_flags(warn_float64='ignore'):
alpha = as_tensor_variable(alpha).astype('float64') alpha = as_tensor_variable(alpha).astype('float64')
assert alpha.ndim == 0 assert alpha.ndim == 0
assert A.ndim == 2 assert A.ndim == 2
...@@ -380,6 +383,7 @@ class GpuGemmBatch(BlasOp): ...@@ -380,6 +383,7 @@ class GpuGemmBatch(BlasOp):
A = as_gpuarray_variable(A, ctx_name) A = as_gpuarray_variable(A, ctx_name)
B = as_gpuarray_variable(B, ctx_name) B = as_gpuarray_variable(B, ctx_name)
C = as_gpuarray_variable(C, ctx_name) C = as_gpuarray_variable(C, ctx_name)
with theano.configparser.change_flags(warn_float64='ignore'):
alpha = as_tensor_variable(alpha).astype('float64') alpha = as_tensor_variable(alpha).astype('float64')
beta = as_tensor_variable(beta).astype('float64') beta = as_tensor_variable(beta).astype('float64')
assert alpha.ndim == 0 assert alpha.ndim == 0
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论