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

Merge pull request #2 from yuxuan-z19/main

Browse Source 
Fix PyTorch 1.13 compilation bug
This commit is contained in:
Lixin Liu 2022-12-02 14:28:07 +08:00 committed by GitHub
commit 8b28ccc512
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
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*
data/cad
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 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});
return {hyperedge_index, hyperedge_list, hyperedge_list_end};