Use theta instead of number of measurements

beomki-yeo · Oct 10, 2024 · 4e247c2 · 4e247c2
1 parent 9a34aeb
commit 4e247c2
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/device/common/include/traccc/fitting/device/impl/fill_sort_keys.ipp b/device/common/include/traccc/fitting/device/impl/fill_sort_keys.ipp
@@ -29,8 +29,8 @@ TRACCC_HOST_DEVICE inline void fill_sort_keys(
     }
 
     // Key = The number of measurements
-    keys_device.at(globalIndex) = static_cast<traccc::scalar>(
-        track_candidates.at(globalIndex).items.size());
+    keys_device.at(globalIndex) =
+        device::get_sort_key(track_candidates.at(globalIndex).header);
     ids_device.at(globalIndex) = globalIndex;
 }