Skip to content

P
pytensor
提交 b420fe59 authored 作者: Frédéric Bastien's avatar Frédéric Bastien 提交者： GitHub
浏览文件
操作

Merge pull request #4830 from abergeron/fix_debugmode

Fix blas C code having wrong alpha and beta for float16.
上级 26445969 3fce8613
隐藏空白字符变更
内嵌 并排
正在显示 包含 7 行增加 和 7 行删除
theano/gpuarray/blas.py
浏览文件 @ b420fe59
...@@ -47,8 +47,8 @@ class GpuGemv(BlasOp): ...@@ -47,8 +47,8 @@ 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)
alpha = as_tensor_variable(alpha) alpha = as_tensor_variable(alpha).astype('float64')
beta = as_tensor_variable(beta) beta = as_tensor_variable(beta).astype('float64')
assert alpha.ndim == 0 assert alpha.ndim == 0
assert beta.ndim == 0 assert beta.ndim == 0
assert A.ndim == 2 assert A.ndim == 2
...@@ -128,8 +128,8 @@ class GpuGemm(BlasOp): ...@@ -128,8 +128,8 @@ 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)
alpha = as_tensor_variable(alpha) alpha = as_tensor_variable(alpha).astype('float64')
beta = as_tensor_variable(beta) beta = as_tensor_variable(beta).astype('float64')
assert alpha.ndim == 0 assert alpha.ndim == 0
assert beta.ndim == 0 assert beta.ndim == 0
assert A.ndim == 2 assert A.ndim == 2
...@@ -208,7 +208,7 @@ class GpuGer(BlasOp): ...@@ -208,7 +208,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)
alpha = as_tensor_variable(alpha) alpha = as_tensor_variable(alpha).astype('float64')
assert alpha.ndim == 0 assert alpha.ndim == 0
assert A.ndim == 2 assert A.ndim == 2
assert x.ndim == 1 assert x.ndim == 1
...@@ -345,8 +345,8 @@ class GpuGemmBatch(BlasOp): ...@@ -345,8 +345,8 @@ 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)
alpha = as_tensor_variable(alpha) alpha = as_tensor_variable(alpha).astype('float64')
beta = as_tensor_variable(beta) beta = as_tensor_variable(beta).astype('float64')
assert alpha.ndim == 0 assert alpha.ndim == 0
assert beta.ndim == 0 assert beta.ndim == 0
assert A.ndim == 3 assert A.ndim == 3
......
Markdown 格式
0% 或
您添加了 0 人 到此讨论。请谨慎行事。
请先完成此评论的编辑！
请 注册 或者 后发表评论