Skip to content

Commit bf73456

Browse files
author
Mark
committed
sign conversion fixes
1 parent 4da11b0 commit bf73456

File tree

2 files changed

+9
-9
lines changed

2 files changed

+9
-9
lines changed

device/cuda/src/gbts_seeding/gbts_seeding_algorithm.cu

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -772,11 +772,11 @@ gbts_seeding_algorithm::output_type gbts_seeding_algorithm::operator()(
772772
cudaStreamSynchronize(stream);
773773

774774
// nPaths to terminus, nTerminusEdges
775-
int path_sizes[2];
775+
unsigned int path_sizes[2];
776776
cudaMemcpyAsync(&path_sizes, &ctx.d_counters[6], 2 * sizeof(unsigned int),
777777
cudaMemcpyDeviceToHost, stream);
778778

779-
TRACCC_DEBUG(path_sizes[0] << "nPath | nTerminus " << path_sizes[1]);
779+
TRACCC_DEBUG(path_sizes[0] << "size of path store | nTerminusEdges " << path_sizes[1]);
780780

781781
cudaMalloc(&ctx.d_path_store,
782782
(path_sizes[0] + path_sizes[1]) * sizeof(int2));
@@ -806,11 +806,11 @@ gbts_seeding_algorithm::output_type gbts_seeding_algorithm::operator()(
806806
kernels::fit_segments<<<nBlocks, nThreads, 0, stream>>>(
807807
ctx.d_reducedSP, ctx.d_output_graph, ctx.d_path_store,
808808
ctx.d_seed_proposals, ctx.d_edge_bids, ctx.d_seed_ambiguity,
809-
ctx.d_levels, ctx.d_counters, path_sizes[0], path_sizes[1],
809+
ctx.d_levels, ctx.d_counters, path_sizes[1],
810810
m_config.minLevel, m_config.max_num_neighbours,
811811
m_config.seed_extraction_params);
812812

813-
int nProps = 0;
813+
unsigned int nProps = 0;
814814
cudaMemcpyAsync(&nProps, &ctx.d_counters[8], sizeof(unsigned int),
815815
cudaMemcpyDeviceToHost, stream);
816816

device/cuda/src/gbts_seeding/kernels/GbtsGraphProcessingKernels.cuh

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -73,7 +73,7 @@ __global__ static void CCA_IterationKernel(const int* d_output_graph,
7373
unsigned int* d_counters, int iter,
7474
unsigned int nEdges,
7575
unsigned int max_num_neighbours,
76-
unsigned int minLevel) {
76+
int minLevel) {
7777

7878
__shared__ unsigned int nEdgesLeft;
7979
unsigned int edge_size = 2 + 1 + max_num_neighbours;
@@ -192,7 +192,7 @@ void __global__ count_terminus_edges(int2* d_path_store,
192192
void __global__ add_terminus_to_path_store(int2* d_path_store,
193193
short2* d_outgoing_paths,
194194
unsigned int* d_counters,
195-
int nEdges) {
195+
unsigned int nEdges) {
196196

197197
int edge_idx = threadIdx.x + blockIdx.x * blockDim.x;
198198
if (edge_idx >= nEdges) {
@@ -226,7 +226,7 @@ void __global__ fill_path_store(int2* d_path_store, int* d_output_graph,
226226

227227
int edge_size = 2 + 1 + max_num_neighbours;
228228

229-
int path_idx = threadIdx.x + blockIdx.x * 16;
229+
unsigned int path_idx = threadIdx.x + blockIdx.x * 16;
230230
// populate live_paths with terminus to start exploration from
231231
if (threadIdx.x < 16 && path_idx < nTerminus) {
232232
int2 path = d_path_store[path_idx];
@@ -558,12 +558,12 @@ void __global__
558558
fit_segments(float4* d_sp_params, int* d_output_graph, int2* d_path_store,
559559
int2* d_seed_proposals, unsigned long long int* d_edge_bids,
560560
char* d_seed_ambiguity, char* d_levels, unsigned int* d_counters,
561-
int nPaths, int nTerminusEdges, int minLevel,
561+
unsigned int nTerminusEdges, int minLevel,
562562
unsigned int max_num_neighbours,
563563
gbts_seed_extraction_params seed_extraction_params) {
564564

565565
// take an extracted path and fit it to produce a quality score
566-
int path_idx = threadIdx.x + blockIdx.x * blockDim.x + nTerminusEdges;
566+
unsigned int path_idx = threadIdx.x + blockIdx.x * blockDim.x + nTerminusEdges;
567567
if (path_idx >= d_counters[7]) {
568568
return;
569569
}

0 commit comments

Comments
 (0)