Skip to content

Commit 81d282c

Browse files
committed
GPU TPC: Tracklet memory during seeding when running on the host should be on the stack
1 parent f1e9522 commit 81d282c

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

GPU/GPUTracking/SectorTracker/GPUTPCTracker.cxx

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -102,9 +102,9 @@ void GPUTPCTracker::RegisterMemoryAllocation()
102102
uint32_t type = GPUMemoryResource::MEMORY_SCRATCH;
103103
if (mRec->GetProcessingSettings().memoryAllocationStrategy == GPUMemoryResource::ALLOCATION_INDIVIDUAL) { // For individual scheme, we allocate tracklets separately, and change the type for the following allocations to custom
104104
type |= GPUMemoryResource::MEMORY_CUSTOM;
105-
mMemoryResTracklets = mRec->RegisterMemoryAllocation(this, &GPUTPCTracker::SetPointersTracklets, type, "TPCTrackerTracklets");
105+
mMemoryResTracklets = mRec->RegisterMemoryAllocation(this, &GPUTPCTracker::SetPointersTracklets, type | GPUMemoryResource::MEMORY_STACK, "TPCTrackerTracklets");
106106
}
107-
mMemoryResOutput = mRec->RegisterMemoryAllocation(this, &GPUTPCTracker::SetPointersOutput, type, "TPCTrackerTracks");
107+
mMemoryResOutput = mRec->RegisterMemoryAllocation(this, &GPUTPCTracker::SetPointersOutput, type, "TPCTrackerTracks"); // TODO: Ideally this should eventually go on the stack, so that we can free it after the first phase of track merging
108108
}
109109

110110
GPUhd() void* GPUTPCTracker::SetPointersTracklets(void* mem)

0 commit comments

Comments
 (0)