Use cuda

Author: n-shevko

examples/benchmark/sparse_vs_dense_tensors.py

@@ -11,7 +11,7 @@
 
 
 assert torch.cuda.is_available(), 'Benchmark works only on cuda'
-device = torch.device("cpu")
+device = torch.device("cuda")
 shape = (500, 500, 500)