Halogenid: Reduktion über eine Domäne für die spezifischen Werte

Question

Ich habe eine Funktion f (x, y, z), in der die Werte entweder 1 und 0 sind, und ich muss die ersten 100 Koordinaten der Werte erhalten, die gleich ist zu 1, zu reduction/update sie auf 0.

Dies ist sehr einfach in c und anderen Sprachen zu realisieren, jedoch habe ich versucht, es mit Halogenid für ein paar Tage zu lösen. Gibt es eine Funktion oder einen Algorithmus, die ich verwenden kann, um es in Halogengeneratoren zu lösen?

Answer 1

Die Frage lautet "Wie implementiere ich Stromverdichtung in Halogenid?" Es wird viel über die parallele Stromverdichtung geschrieben und es ist etwas nicht triviales, gut zu machen. Sehen Sie diesen Stack-Überlauf Antwort darauf in CUDA für einige Diskussion zu tun und Referenzen: CUDA stream compaction algorithm

Eine schnelle Realisierung von einfachen Strom Verdichtung in Halide eine Präfixsumme mit sieht aus wie so:

#include "Halide.h" 
#include <iostream> 

using namespace Halide; 

static void print_1d(const Buffer<int32_t> &result) { 
    std::cout << "{ "; 
    const char *prefix = ""; 
    for (int i = 0; i < result.dim(0).extent(); i++) { 
     std::cout << prefix << result(i); 
     prefix = ", "; 
    } 
    std::cout << "}\n"; 

} 

int main(int argc, char **argv) { 
    uint8_t vals[] = {0, 10, 99, 76, 5, 200, 88, 15}; 
    Buffer<uint8_t> in(vals); 

    Var x; 
    Func prefix_sum; 

    RDom range(1, in.dim(0).extent() - 1); 
    prefix_sum(x) = (int32_t)0; 
    prefix_sum(range) = select(in(range - 1) > 42, prefix_sum(range - 1) + 1, prefix_sum(range - 1)); 

    RDom in_range(0, in.dim(0).extent()); 
    Func compacted_indices; 
    compacted_indices(x) = -1; 
    compacted_indices(clamp(prefix_sum(in_range), 0, in.dim(0).extent() - 1)) = select(in(in_range) > 42, in_range, - 1); 

    Buffer<int32_t> sum = prefix_sum.realize(8); 
    Buffer<int32_t> indices = compacted_indices.realize(8); 

    print_1d(sum); 
    print_1d(indices); 

    return 0; 
}