from nose.plugins import attrib

gpu = attrib.attr('gpu')


def multi_gpu(gpu_num):
    return attrib.attr(gpu=gpu_num)

@@ -51,16 +51,52 @@
 ('gpu')%0A
+cudnn = attrib.attr('gpu', 'cudnn')%0A
 %0A%0Adef mu

Fix test