cuda_skip = { "TestArray.test_api", "TestArray.test_setitem", "TestAutograd.test_cumprod_grad", "TestAutograd.test_slice_grads", "TestAutograd.test_split_against_slice", "TestAutograd.test_stop_gradient", "TestAutograd.test_topk_grad", "TestAutograd.test_update_state", "TestAutograd.test_vjp", "TestBF16.test_arg_reduction_ops", "TestBF16.test_binary_ops", "TestBF16.test_reduction_ops", "TestBlas.test_block_masked_matmul", "TestBlas.test_complex_gemm", "TestBlas.test_gather_matmul", "TestBlas.test_gather_matmul_grad", "TestBlas.test_matmul_batched", "TestBlas.test_matrix_vector_attn", "TestCompile.test_compile_dynamic_dims", "TestCompile.test_compile_inf", "TestCompile.test_inf_constant", "TestConv.test_1d_conv_with_2d", "TestConv.test_asymmetric_padding", "TestConv.test_basic_grad_shapes", "TestConv.test_conv2d_unaligned_channels", "TestConv.test_conv_1d_groups_flipped", "TestConv.test_conv_general_flip_grad", "TestConv.test_conv_groups_grad", "TestConv.test_numpy_conv", "TestConv.test_repeated_conv", "TestConv.test_torch_conv_1D", "TestConv.test_torch_conv_1D_grad", "TestConv.test_torch_conv_2D", "TestConv.test_torch_conv_2D_grad", "TestConv.test_torch_conv_3D", "TestConv.test_torch_conv_3D_grad", "TestConv.test_torch_conv_depthwise", "TestConv.test_torch_conv_general", "TestConvTranspose.test_torch_conv_tranpose_1d_output_padding", "TestConvTranspose.test_torch_conv_transpose_1D", "TestConvTranspose.test_torch_conv_transpose_1D_grad", "TestConvTranspose.test_torch_conv_transpose_2D", "TestConvTranspose.test_torch_conv_transpose_2D_grad", "TestConvTranspose.test_torch_conv_transpose_2d_output_padding", "TestConvTranspose.test_torch_conv_transpose_3D", "TestConvTranspose.test_torch_conv_transpose_3D_grad", "TestConvTranspose.test_torch_conv_transpose_3d_output_padding", "TestEinsum.test_attention", "TestEinsum.test_ellipses", "TestEinsum.test_opt_einsum_test_cases", "TestEval.test_multi_output_eval_during_transform", "TestExportImport.test_export_conv", "TestFast.test_rope_grad", "TestFFT.test_fft", "TestFFT.test_fft_big_powers_of_two", "TestFFT.test_fft_contiguity", "TestFFT.test_fft_exhaustive", "TestFFT.test_fft_grads", "TestFFT.test_fft_into_ifft", "TestFFT.test_fft_large_numbers", "TestFFT.test_fft_shared_mem", "TestFFT.test_fftn", "TestInit.test_orthogonal", "TestLinalg.test_cholesky", "TestLinalg.test_cholesky_inv", "TestLinalg.test_eig", "TestLinalg.test_eigh", "TestLinalg.test_inverse", "TestLinalg.test_lu", "TestLinalg.test_lu_factor", "TestLinalg.test_pseudo_inverse", "TestLinalg.test_qr_factorization", "TestLinalg.test_svd_decomposition", "TestLinalg.test_tri_inverse", "TestLoad.test_load_f8_e4m3", "TestLosses.test_binary_cross_entropy", "TestMemory.test_memory_info", "TestLayers.test_conv1d", "TestLayers.test_conv2d", "TestLayers.test_elu", "TestLayers.test_group_norm", "TestLayers.test_hard_shrink", "TestLayers.test_pooling", "TestLayers.test_quantized_embedding", "TestLayers.test_sin_pe", "TestLayers.test_softshrink", "TestLayers.test_upsample", "TestOps.test_argpartition", "TestOps.test_array_equal", "TestOps.test_as_strided", "TestOps.test_atleast_1d", "TestOps.test_atleast_2d", "TestOps.test_atleast_3d", "TestOps.test_binary_ops", "TestOps.test_bitwise_grad", "TestOps.test_complex_ops", "TestOps.test_divmod", "TestOps.test_dynamic_slicing", "TestOps.test_hadamard", "TestOps.test_hadamard_grad_vmap", "TestOps.test_irregular_binary_ops", "TestOps.test_kron", "TestOps.test_log", "TestOps.test_log10", "TestOps.test_log1p", "TestOps.test_log2", "TestOps.test_logaddexp", "TestOps.test_logcumsumexp", "TestOps.test_partition", "TestOps.test_scans", "TestOps.test_slice_update_reversed", "TestOps.test_softmax", "TestOps.test_sort", "TestOps.test_tensordot", "TestOps.test_tile", "TestQuantized.test_gather_matmul_grad", "TestQuantized.test_gather_qmm", "TestQuantized.test_gather_qmm_sorted", "TestQuantized.test_non_multiples", "TestQuantized.test_qmm", "TestQuantized.test_qmm_jvp", "TestQuantized.test_qmm_shapes", "TestQuantized.test_qmm_vjp", "TestQuantized.test_qmv", "TestQuantized.test_quantize_dequantize", "TestQuantized.test_qvm", "TestQuantized.test_qvm_splitk", "TestQuantized.test_small_matrix", "TestQuantized.test_throw", "TestQuantized.test_vjp_scales_biases", "TestReduce.test_axis_permutation_sums", "TestReduce.test_dtypes", "TestReduce.test_expand_sums", "TestReduce.test_many_reduction_axes", "TestUpsample.test_torch_upsample", "TestVmap.test_vmap_conv", "TestVmap.test_vmap_inverse", "TestVmap.test_vmap_svd", }