use cuda if available (in build_one_tensor_batch)

5 jobs for master
Status Name Job ID Coverage
  Build
canceled build #428

 
  Test
canceled bandit-sast #431
allowed to fail

canceled code_quality #430
allowed to fail

canceled secret_detection_default_branch #432

canceled test #429