Fix CI Error

emailweixu-patch-1
Yang Yu 7 years ago
parent adf14b0c70
commit e3b0af43a0

@ -228,7 +228,7 @@ class TestLayerNormdOp(OpTest):
place)
places = [core.CPUPlace()]
if core.is_compile_gpu() and core.op_support_gpu("layer_norm"):
if core.is_compiled_with_cuda() and core.op_support_gpu("layer_norm"):
places.append(core.CUDAPlace(0))
for place in places:

Loading…
Cancel
Save