detrex icon indicating copy to clipboard operation
detrex copied to clipboard

Published 20 hours ago • verified publisher icon IDEA-Research

Skip gpu-only tests if cuda is not available

Open dgcnz opened this issue 7 months ago • 0 comments

Changes:

  • [x] Skip tests/test_ms_deform_attn.py when torch.cuda.is_available() yields false.
  • [x] Move cuda() calls from global scope to the local scope of tests.

dgcnz avatar Jul 10 '24 08:07 dgcnz