Commit e6638edd authored by rusty1s's avatar rusty1s
Browse files

added codecov

parent c663c5b1
import time
import torch
from torch_cluster._ext import ffi
cluster = torch.cuda.LongTensor(4)
row = torch.cuda.LongTensor([0, 0, 1, 1, 1, 2, 2, 2, 3, 3])
col = torch.cuda.LongTensor([1, 2, 0, 2, 3, 0, 1, 3, 1, 2])
# deg = torch.cuda.LongTensor([2, 3, 3, 2])
func = ffi.THCCGreedy
print(func)
a = 0
torch.cuda.synchronize()
t = time.perf_counter()
# for i in range(100):
func(cluster, row, col)
# a += cluster.sum() / cluster.size(0)
torch.cuda.synchronize()
print(time.perf_counter() - t)
print(cluster)
......@@ -5,4 +5,4 @@ description-file = README.md
test = pytest
[tool:pytest]
addopts = --capture=no
addopts = --capture=no --cov
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment