Jan 16, 2025 · Jan 16, 2025 · Jan 16, 2025 · Jan 17, 2025 · Jan 17, 2025 · Jan 16, 2025
diff --git a/test/test_tensor_creation_ops.py b/test/test_tensor_creation_ops.py
                                                            sparse_size, dtype=torch.float64)
                self.assertEqual(sparse_with_dtype.device, torch.device('cpu'))

    @onlyCUDA
    @onlyNativeDeviceTypes
    def test_new_tensor_device(self, device):
        torch_device = torch.device(device)
        cpu_device = torch.device('cpu')
        tensor = torch.tensor((1, 2, 3), device=device)

        # need more than one device_type to test this
        assert self.device_type == 'cuda'

        with warnings.catch_warnings():
            warnings.simplefilter("ignore")

            self.assertEqual(tensor.new_tensor(tensor.cpu()).device, torch_device)
            self.assertEqual(tensor.cpu().new_tensor(tensor).device, cpu_device)
            self.assertEqual(tensor.new_tensor(tensor, device=cpu_device).device, cpu_device)

    def _test_signal_window_functions(self, name, dtype, device, **kwargs):
        import scipy.signal as signal
Original file line number	Diff line number	Diff line change
Expand Up		@@ -2758,6 +2758,23 @@ def test_tensor_ctor_device_inference(self, device):
		sparse_size, dtype=torch.float64)
		self.assertEqual(sparse_with_dtype.device, torch.device('cpu'))

		@onlyCUDA
		@onlyNativeDeviceTypes
		def test_new_tensor_device(self, device):
		torch_device = torch.device(device)
		cpu_device = torch.device('cpu')
		tensor = torch.tensor((1, 2, 3), device=device)

		# need more than one device_type to test this
		assert self.device_type == 'cuda'

		with warnings.catch_warnings():
		warnings.simplefilter("ignore")

		self.assertEqual(tensor.new_tensor(tensor.cpu()).device, torch_device)
		self.assertEqual(tensor.cpu().new_tensor(tensor).device, cpu_device)
		self.assertEqual(tensor.new_tensor(tensor, device=cpu_device).device, cpu_device)

		def _test_signal_window_functions(self, name, dtype, device, **kwargs):
		import scipy.signal as signal

Expand Down