lxchen/Xplace_for_ICCAD
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix on cuda117

Browse Source
This commit is contained in:
yuxuan-z19 2022-12-02 13:33:57 +08:00
parent 00ea219123
commit 1675fe458f
2 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
.gitignore vendored
View File

@ -13,3 +13,4 @@ build/
result* result*
data/cad data/cad
misc misc
.venv/

2
cpp_to_py/io_parser/gp/GPDatabase.cpp
View File

@ -487,7 +487,7 @@ std::vector<torch::Tensor> GPDatabase::getHyperedgeInfoTensor() {
} }
auto hyperedge_index = torch::cat({hyperedge_list.unsqueeze(0), hyperedge_index_helper.unsqueeze(0)}, 0); auto hyperedge_index = torch::cat({hyperedge_list.unsqueeze(0), hyperedge_index_helper.unsqueeze(0)}, 0);
auto new_order_idx = torch::argsort(hyperedge_index.index({0}), 0); auto new_order_idx = torch::argsort(hyperedge_index.index({0}), 0, false);
hyperedge_index = hyperedge_index.index({torch::indexing::Slice(), new_order_idx}); hyperedge_index = hyperedge_index.index({torch::indexing::Slice(), new_order_idx});
return {hyperedge_index, hyperedge_list, hyperedge_list_end}; return {hyperedge_index, hyperedge_list, hyperedge_list_end};