add device process

zty-king · zty-king · commit 62f3c7de9921 · 2025-11-18T09:02:51.000Z
diff --git a/python/paddle/audio/functional/window.py b/python/paddle/audio/functional/window.py
@@ -493,7 +493,7 @@ def _apply_window_postprocess(
                 raise RuntimeError(
                     f"Pinning memory is not supported for {device}"
                 )
-
+    w = paddle.to_tensor(w, place=device)
     if pin_memory and paddle.in_dynamic_mode():
         w = w.pin_memory()
     if requires_grad is True:
diff --git a/test/legacy_test/CMakeLists.txt b/test/legacy_test/CMakeLists.txt
@@ -876,7 +876,6 @@ set_tests_properties(test_cross_entropy_loss PROPERTIES TIMEOUT 180)
 set_tests_properties(test_legacy_loss_args PROPERTIES TIMEOUT 10)
 set_tests_properties(test_activation_nn_grad PROPERTIES TIMEOUT 250)
 set_tests_properties(test_empty_op PROPERTIES TIMEOUT 120)
-set_tests_properties(test_window PROPERTIES TIMEOUT 10)
 set_tests_properties(test_elementwise_div_op PROPERTIES TIMEOUT 120)
 set_tests_properties(test_multiclass_nms_op PROPERTIES TIMEOUT 120)
 if(NOT WIN32)

Original file line number	Diff line number	Diff line change
`@@ -493,7 +493,7 @@ def _apply_window_postprocess(`
`493`	`493`	`raise RuntimeError(`
`494`	`494`	`f"Pinning memory is not supported for {device}"`
`495`	`495`	`)`
`496`		`-`
	`496`	`+ w = paddle.to_tensor(w, place=device)`
`497`	`497`	`if pin_memory and paddle.in_dynamic_mode():`
`498`	`498`	`w = w.pin_memory()`
`499`	`499`	`if requires_grad is True:`