[rust] Fixes test failure on GPU

[frankfliu]

Description

Brief description of what this PR is about

• If this change is a backward incompatible change, why must this change be made?
• Interesting edge cases to note here

[codecov-commenter]

⚠️ Please install the to ensure uploads and comments are reliably processed by Codecov.

Codecov Report

All modified and coverable lines are covered by tests ✅

Project coverage is 72.70%. Comparing base (725851b) to head (4dd1cb2).
Report is 141 commits behind head on master.

❗ Your organization needs to install the Codecov GitHub app to enable full functionality.

Additional details and impacted files

@@             Coverage Diff              @@
##             master    #3301      +/-   ##
============================================
- Coverage     72.71%   72.70%   -0.01%     
+ Complexity     7361     7359       -2     
============================================
  Files           677      677              
  Lines         32561    32561              
  Branches       3433     3433              
============================================
- Hits          23677    23675       -2     
- Misses         7258     7261       +3     
+ Partials       1626     1625       -1     

☔ View full report in Codecov by Sentry.
📢 Have feedback on the report? Share it here.