[Bug]: unit test failed on gpu

[gufengc]

Steps to Reproduce

run pytest on nvidia gpu

Expected Behavior

pass

Actual Behavior

fail

Version

main

Environment & Context

• I'm using the latest version.
• I have searched existing issues.

[gufengc]

Consider add github action to run unit test on nvidia gpu