test/test_indexing_cuda.py

import torch
from test_indexing import *


if __name__ == '__main__':
    if torch.cuda.is_available():
        torch.set_default_tensor_type(torch.cuda.FloatTensor)
        run_tests()
    else:
        print("Skipping test_indexing_cuda.py")