Skip to content

Commit ad8e2a9

Browse files
authored
fix: Zero remaining edges in TorchEdgeClassifier (acts-project#4325)
<!-- This is an auto-generated comment: release notes by coderabbit.ai --> ## Summary by CodeRabbit - **Bug Fixes** - Improved error handling by detecting and reporting cases where no edges remain after filtering, preventing further processing in such scenarios. <!-- end of auto-generated comment: release notes by coderabbit.ai -->
1 parent 0eaaefd commit ad8e2a9

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

Plugins/ExaTrkX/src/TorchEdgeClassifier.cpp

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -170,6 +170,10 @@ PipelineTensors TorchEdgeClassifier::operator()(
170170
torch::Tensor edgesAfterCut = edgeIndex.index({Slice(), mask});
171171
edgesAfterCut = edgesAfterCut.to(torch::kInt64);
172172

173+
if (edgesAfterCut.numel() == 0) {
174+
throw NoEdgesError{};
175+
}
176+
173177
ACTS_VERBOSE("Size after score cut: " << edgesAfterCut.size(1));
174178
printCudaMemInfo(logger());
175179
t3 = std::chrono::high_resolution_clock::now();

0 commit comments

Comments
 (0)