File tree Expand file tree Collapse file tree 1 file changed +0
-19
lines changed
ivy_tests/test_ivy/test_frontends/test_torch Expand file tree Collapse file tree 1 file changed +0
-19
lines changed Original file line number Diff line number Diff line change @@ -5583,25 +5583,6 @@ def test_torch_cross(
55835583 )
55845584
55855585
5586- @given (
5587- dtype_x = helpers .dtype_and_values (
5588- available_dtypes = helpers .get_dtypes ("valid" , prune_function = False ),
5589- ).filter (
5590- lambda x : "bfloat16" not in x [0 ]
5591- and "uint16" not in x [0 ]
5592- and "uint32" not in x [0 ]
5593- and "uint64" not in x [0 ]
5594- ),
5595- )
5596- def test_torch_cuda (dtype_x , backend_fw ):
5597- ivy .set_backend (backend_fw )
5598- _ , data = dtype_x
5599- x = Tensor (data [0 ], device = "gpu:0" )
5600- device = "gpu:0"
5601- ivy .utils .assertions .check_equal (x .cuda , device , as_array = False )
5602- ivy .previous_backend ()
5603-
5604-
56055586# cummax
56065587@handle_frontend_method (
56075588 class_tree = CLASS_TREE ,
You can’t perform that action at this time.
0 commit comments