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;
}
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;
}
