forked from Adam27X/hybrid_BC
-
Notifications
You must be signed in to change notification settings - Fork 0
/
kernels.cuh
19 lines (14 loc) · 838 Bytes
/
kernels.cuh
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
#ifndef BC_KERNELS
#define BC_KERNELS
#include <vector>
#include <cuda.h>
#include <cstdio>
#include <iostream>
#include <thrust/copy.h>
#include <thrust/device_vector.h>
#include "parse.h"
#include "util.cuh"
__device__ void bitonic_sort(int *values, int N);
__global__ void bc_gpu_opt(float *bc, const int *R, const int *C, const int *F, const int n, const int m, int *d, unsigned long long *sigma, float *delta, int *Q, int *Q2, int *S, int *endpoints, int *next_source, size_t pitch_d, size_t pitch_sigma, size_t pitch_delta, size_t pitch_Q, size_t pitch_Q2, size_t pitch_S, size_t pitch_endpoints, int start, int end, int *jia, int *diameters, int *source_vertices, bool approx);
std::vector<float> bc_gpu(graph g, int max_threads_per_block, int number_of_SMs, program_options op, const std::set<int> &source_vertices);
#endif