Skip to content

Commit ed217ee

Browse files
committed
refactor: remove test_torch_get_device
1 parent 80b2598 commit ed217ee

File tree

1 file changed

+0
-25
lines changed

1 file changed

+0
-25
lines changed

ivy_tests/test_ivy/test_frontends/test_torch/test_tensor.py

Lines changed: 0 additions & 25 deletions
Original file line numberDiff line numberDiff 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-
74917466
def test_torch_grad(backend_fw):
74927467
ivy.set_backend(backend_fw)
74937468
x = Tensor(ivy.array([1.0, 2.0, 3.0]))

0 commit comments

Comments
 (0)