torchopt icon indicating copy to clipboard operation
torchopt copied to clipboard
verified publisher icon metaopt

[BUG] Accelerated optimizer test failed on GPU

Open Benjamin-eecs opened this issue 3 years ago • 1 comments

Describe the bug

CUDA part test failed on GPU machine

Screenshots

torchopt_bug_2

System info

torchopt_bug_1

import torchopt, numpy, sys
print(torchopt.__version__, numpy.__version__, sys.version, sys.platform)
0.4.1 1.22.3 3.8.13 (default, Mar 28 2022, 11:38:47) 
[GCC 7.5.0] linux

Checklist

  • [x] I have checked that there is no similar issue in the repo (required)
  • [x] I have read the documentation (required)

Benjamin-eecs avatar Jul 18 '22 17:07 Benjamin-eecs

Reproduced with NVIDIA A100 GPU:

image

XuehaiPan avatar Aug 05 '22 07:08 XuehaiPan