Skip to content

Commit

Permalink
update
Browse files Browse the repository at this point in the history
  • Loading branch information
rusty1s committed Apr 28, 2024
1 parent 1fa1dc4 commit 56b40f6
Show file tree
Hide file tree
Showing 2 changed files with 3 additions and 3 deletions.
2 changes: 1 addition & 1 deletion .github/workflows/cpp_testing.yml
Original file line number Diff line number Diff line change
Expand Up @@ -23,7 +23,7 @@ jobs:
- name: Configure
run: |
export MKL_VERSION=`python -c 'from tools.mkl_ver import compatible_mkl_ver;print(compatible_mkl_ver())'`
pip install mkl-include==${MKL_VERSION} mkl-static==${MKL_VERSION}
pip install --no-cache-dir mkl-include==${MKL_VERSION} mkl-static==${MKL_VERSION}
export _BLAS_INCLUDE_DIR=`python -c 'import os;import sysconfig;data=sysconfig.get_path("data");print(f"{data}{os.sep}include")'`
export MKL_DIR=`python -c 'import os;import sysconfig;data=sysconfig.get_path("data");print(f"{data}{os.sep}lib{os.sep}cmake{os.sep}mkl")'`
Expand Down
4 changes: 2 additions & 2 deletions benchmark/sampler/subgraph.py
Original file line number Diff line number Diff line change
Expand Up @@ -16,15 +16,15 @@ def test_subgraph(dataset, **kwargs):
t = time.perf_counter()
for _ in range(10):
pyg_lib.sampler.subgraph(rowptr, col, nodes)
print(f'time={time.perf_counter()-t:.6f} seconds')
print(f'time={time.perf_counter() - t:.6f} seconds')

edge_index = to_edge_index(rowptr, col)
from torch_geometric.utils import subgraph

t = time.perf_counter()
for _ in range(10):
subgraph(nodes, edge_index, num_nodes=num_nodes, relabel_nodes=True)
print(f'time={time.perf_counter()-t:.6f} seconds')
print(f'time={time.perf_counter() - t:.6f} seconds')


if __name__ == '__main__':
Expand Down

0 comments on commit 56b40f6

Please sign in to comment.