dmlc--dgl
czkkkkkk
7c059e86c7
* [Sparse] Add sparse matrix C++ implementation * Add documentation * Update * Minor fix * Move Python code to dgl/mock_sparse2 * Move headers to include * lint * Update * Add dgl_sparse directory * Move src code to dgl_sparse * Add __init__.py in tests to avoid naming conflict * Add dgl sparse so in Jenkinsfile * Complete docstring & SparseMatrix basic op * lint * Disable win tests
12 行
255 B
Python
12 行
255 B
Python
import os
|
|
|
|
import torch
|
|
|
|
cmake_prefix_path = getattr(
|
|
torch.utils,
|
|
"cmake_prefix_path",
|
|
os.path.join(os.path.dirname(torch.__file__), "share", "cmake"),
|
|
)
|
|
version = torch.__version__.split("+")[0]
|
|
print(";".join([cmake_prefix_path, version]))
|