mirror of https://github.com/llvm/torch-mlir
[torch-mlir][sparse] fixed merge conflict (#2967)
parent
e85a2a87c5
commit
f21b76b68a
|
@ -292,7 +292,6 @@ def test_sparse_SpMM():
|
||||||
print(m)
|
print(m)
|
||||||
|
|
||||||
# Run it with PyTorch torch.sparse and with TORCH-MLIR sparse_jit.
|
# Run it with PyTorch torch.sparse and with TORCH-MLIR sparse_jit.
|
||||||
net = MatMulNet()
|
|
||||||
res1 = net(sparse_input, dense_input)
|
res1 = net(sparse_input, dense_input)
|
||||||
res2 = sparse_jit(net, sparse_input, dense_input)
|
res2 = sparse_jit(net, sparse_input, dense_input)
|
||||||
print("torch.sparse")
|
print("torch.sparse")
|
||||||
|
|
Loading…
Reference in New Issue