提交 5a180d18 authored 作者: Boris Fomitchev's avatar Boris Fomitchev 提交者: notoraptor

fixing flake8

上级 35bb0013
......@@ -1204,7 +1204,7 @@ __device__ ga_half atomicExch(ga_half *addr, ga_half val) {
new_ = __byte_perm(old, __HALF_TO_US(val), ((ga_size)addr & 2) ? 0x5410 : 0x3254);
old = atomicCAS(base, assumed, new_);
} while (assumed != old);
ga_half ret;
ga_half ret;
__HALF_TO_US(ret) =__byte_perm(old, 0,
((ga_size)addr & 2) ? 0x4432 : 0x4410);
return ret;
......
......@@ -2298,6 +2298,7 @@ class Cudnn_grouped_conv(Grouped_conv_noOptim):
flip_filter = False
is_dnn = True
class Cudnn_grouped_conv3d(Grouped_conv3d_noOptim):
mode = mode_with_gpu
conv = staticmethod(dconvfwd)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论