提交 0b2f4053 authored 作者: xiaoqie's avatar xiaoqie

fix type

上级 ea0aff00
......@@ -597,7 +597,7 @@ class GpuSoftmax(GpuKernelBase, Op):
type_sm = gpuarray.dtype_to_ctype(dtype_sm)
type_acc = gpuarray.dtype_to_ctype(work_sm)
ctype = gpuarray.dtype_to_ctype(dtype_sm)
ctype = gpuarray.dtype_to_ctype(work_sm)
params = [
gpuarray.SIZE, gpuarray.SIZE,
......@@ -915,7 +915,7 @@ class GpuSoftmaxWithBias(GpuKernelBase, Op):
type_sm = gpuarray.dtype_to_ctype(dtype_sm)
type_acc = gpuarray.dtype_to_ctype(work_sm)
ctype = gpuarray.dtype_to_ctype(dtype_sm)
ctype = gpuarray.dtype_to_ctype(work_sm)
params = [
gpuarray.SIZE, gpuarray.SIZE,
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论