auto_count.hpp

Go to the documentation of this file.

// Copyright (c) 2021 - 2023 SparseLinearAlgebra
// Autogenerated file, do not modify
 
#pragma once
 
static const char source_count[] = R"(
 
 
__kernel void count_nz(__global const TYPE* g_vec,
                       __global uint*       g_count,
                       const uint           n) {
    const uint gid   = get_global_id(0);
    const uint gsize = get_global_size(0);
    uint       count = 0;
 
    for (uint i = gid; i < n; i += gsize) {
        if (g_vec[i] != 0) {
            count += 1;
        }
    }
 
    atomic_add(g_count, count);
}
 
__kernel void count_mf(__global const TYPE* g_vec,
                       __global uint*       g_count,
                       const uint           n,
                       const TYPE           ref) {
    const uint gid   = get_global_id(0);
    const uint gsize = get_global_size(0);
    uint       count = 0;
 
    for (uint i = gid; i < n; i += gsize) {
        if (g_vec[i] != ref) {
            count += 1;
        }
    }
 
    atomic_add(g_count, count);
}
)";