macro_rules! test_device {
    ($fn_name: ident, $test_cpu: ident, $test_cuda: ident) => { ... };
}