nvGRAPH三角形计数和遍历示例

nvGRAPH三角形计数和遍历示例

#include “ stdlib.h”

#include“ inttypes.h”

#include“ stdio.h”

#include“ nvgraph.h”

#define check( a )\

{\

nvgraphStatus_t status =（a）; \

if（（status）！= NVGRAPH_STATUS_SUCCESS）{\

        printf（“ERROR ：％s，％s：％d \ n”，status，__ FILE __，__ LINE __）； \

exit（0）; \

} \

}

int main（int argc，char ** argv）

{

// nvgraph变量

nvgraphHandle_t handle；

nvgraphGraphDescr_t graph;

nvgraphCSRTopology32I_t CSR_input;

//初始化主机数据

CSR_input =（nvgraphCSRTopology32I_t）malloc（sizeof（struct nvgraphCSRTopology32I_st））;

//无向 graph：

// 3个三角形

//邻接矩阵下三角形的CSR：

const size_t n = 6，nnz = 8;

int source_offsets [] = {0，0，1，2，4，4，6，8};

int destination_indices [] = {0，1，1，2，2，2，3，3，4};

check（nvgraphCreate（＆handle））;

check（nvgraphCreateGraphDescr（ handle＆graph））;

CSR_input-> nvertices = n;

CSR_input-> nedges = nnz;

CSR_input-> source_offsets = source_offsets;

CSR_input-> destination_indices = destination_indices;

//设置 graph连接性

check（nvgraphSetGraphStructure（handle，graph，（void *）CSR_input，NVGRAPH_CSR_32））;

uint64_t trcount = 0;

check（nvgraphTriangleCount（ handle， graph，＆trcount））;

printf（“三角形数：％” PRIu64 “ \ n”，trcount）;

free（CSR_input）;

check（nvgraphDestroyGraphDescr（ handle， graph））;

check（nvgraphDestroy（ handle））;

return 0;

}

nvGRAPH遍历示例

void check_status（nvgraphStatus_t status）{

if（（int）status！= 0）{

printf（“error：％d \ n”，status）；

exit（0）;

}

}

int main（int argc，char ** argv）{

// graph示例（CSR格式）

const size_t n = 7，nnz = 12，vertex_numsets = 2，edge_numset = 0;

int source_offsets_h [] = {0，1，3，4，6，6，8，10，12};

int destination_indices_h [] = {5，0，2，0，4，5，5，2，3，3，4，1，5};

//存储结果的位置（与源的距离）和存储结果的位置（搜索树中的前身）

int bfs_distances_h [n]，bfs_predecessors_h [n];

// nvgraph变量

nvgraphStatus_tstatus；

nvgraphHandle_t handle；

nvgraphGraphDescr_t graph;

nvgraphCSRTopology32I_t CSR_input;

cudaDataType_t * vertex_dimT;

size_t distances_index = 0;

size_t predecessors_index = 1;

vertex_dimT =（cudaDataType_t *）malloc（vertex_numsets * sizeof（cudaDataType_t））;

vertex_dimT [distances_index] = CUDA_R_32I;

vertex_dimT [predecessors_index] = CUDA_R_32I;

//创建nvgraph对象

check_status（nvgraphCreate（＆handle））;

check_status（nvgraphCreateGraphDescr（ handle＆graph））;

//设置 graph的连通性和属性（转移）

CSR_input =（nvgraphCSRTopology32I_t）malloc（sizeof（struct nvgraphCSCTopology32I_st））;

CSR_input-> nvertices = n;

CSR_input-> nedges = nnz;

CSR_input-> source_offsets = source_offsets_h;

CSR_input-> destination_indices = destination_indices_h;

check_status（nvgraphSetGraphStructure（handle，graph，（void *）CSR_input，NVGRAPH_CSR_32））;;

check_status（nvgraphAllocateVertexData（ handle， graph，vertex_numsets，vertex_dimT））;

int source_vert = 1;

//设置遍历参数

nvgraphTraversalParameter_t traversal_param;

nvgraphTraversalParameterInit（＆traversal_param）;

nvgraphTraversalSetDistancesIndex（＆traversal_param，distances_index）;

nvgraphTraversalSetPredecessorsIndex（＆traversal_param，predecessors_index）;

nvgraphTraversalSetUndirectedFlag（＆traversal_param，false）;

//使用BFS算法进行遍历

check_status（nvgraphTraversal（ handle， graph，NVGRAPH_TRAVERSAL_BFS，＆source_vert，traversal_param））;

//获取结果

check_status（nvgraphGetVertexData（handle， graph表，（void *）bfs_distances_h，distances_index））;

check_status（nvgraphGetVertexData（ handle， graph，（void *）bfs_predecessors_h，predecessors_index））;

//

for（int i = 0; i <n; i ++），期望bfs distances_h =（1 0 1 3 3 2 2147483647） printf（“距顶点％d的距离：％i \ n”，i，bfs_distances_h [i]） ; printf（“ \ n”）;

//

for（int i = 0; i <n; i ++），期望bfs前驱体=（1 -1 1 5 5 0 -1） printf（“顶点％d的前驱体：％i \ n”，i，bfs_predecessors_h [i ]）; printf（“ \ n”）;

free（vertex_dimT）;

free（CSR_input）;

check_status（nvgraphDestroyGraphDescr（ handle， graph））;

check_status（nvgraphDestroy（handle））;

return 0;

}