File tree Expand file tree Collapse file tree 1 file changed +0
-25
lines changed
ivy_tests/test_ivy/test_frontends/test_torch Expand file tree Collapse file tree 1 file changed +0
-25
lines changed Original file line number Diff line number Diff line change @@ -7463,31 +7463,6 @@ def test_torch_gcd(
74637463 )
74647464
74657465
7466- @given (
7467- dtype_x = helpers .dtype_and_values (
7468- available_dtypes = helpers .get_dtypes ("valid" , prune_function = False )
7469- ).filter (
7470- lambda x : "bfloat16" not in x [0 ]
7471- and "uint16" not in x [0 ]
7472- and "uint32" not in x [0 ]
7473- and "uint64" not in x [0 ]
7474- ),
7475- )
7476- def test_torch_get_device (
7477- dtype_x ,
7478- backend_fw ,
7479- ):
7480- ivy .set_backend (backend_fw )
7481- _ , data = dtype_x
7482- x = Tensor (data [0 ])
7483- ivy .utils .assertions .check_equal (x .get_device , - 1 , as_array = False )
7484- x = Tensor (data [0 ], "gpu:0" )
7485- ivy .utils .assertions .check_equal (x .get_device , 0 , as_array = False )
7486- x = Tensor (data [0 ], "tpu:3" )
7487- ivy .utils .assertions .check_equal (x .get_device , 3 , as_array = False )
7488- ivy .previous_backend ()
7489-
7490-
74917466def test_torch_grad (backend_fw ):
74927467 ivy .set_backend (backend_fw )
74937468 x = Tensor (ivy .array ([1.0 , 2.0 , 3.0 ]))
You can’t perform that action at this time.
0 commit comments