From 431c871af76c7fd4c20c20f92ce295e8dd513102 Mon Sep 17 00:00:00 2001 From: Stanislaw Adaszewski Date: Fri, 17 Jul 2020 12:58:39 +0200 Subject: [PATCH] Add test_timing_05(). --- .gitignore | 1 + tests/icosagon/test_trainloop.py | 13 +++++++++++++ 2 files changed, 14 insertions(+) diff --git a/.gitignore b/.gitignore index eb03072..9ec7cab 100644 --- a/.gitignore +++ b/.gitignore @@ -4,3 +4,4 @@ __pycache__ /docs/icosagon/*.dot /docs/icosagon/*.png /experiments/decagon_run/profiler_results +/experiments/decagon_run_effcat/profiler_results diff --git a/tests/icosagon/test_trainloop.py b/tests/icosagon/test_trainloop.py index accbea9..c052dba 100644 --- a/tests/icosagon/test_trainloop.py +++ b/tests/icosagon/test_trainloop.py @@ -112,3 +112,16 @@ def test_timing_04(): for _ in range(1300): _ = torch.sparse.mm(adj_mat, rep) print('Elapsed:', time.time() - t) + + +def test_timing_05(): + if torch.cuda.device_count() == 0: + pytest.skip('Test requires CUDA') + dev = torch.device('cuda:0') + adj_mat = (torch.rand(2000, 2000) < .001).to(torch.float32).to_sparse().to(dev) + rep = torch.eye(2000).requires_grad_(True).to(dev) + t = time.time() + for _ in range(1300): + _ = torch.sparse.mm(adj_mat, rep) + torch.cuda.synchronize() + print('Elapsed:', time.time() - t)