스러스트 반복자 믹스 사용법

Question

#include <stdio.h> 
#include <stdlib.h> 
#include <cuda.h> 
#include <thrust/device_vector.h> 
#include <thrust/host_vector.h> 
#include <thrust/scan.h> 
#include <thrust/execution_policy.h> 
#include <iostream> 
#include <thrust/transform.h> 
struct text_functor { 
    text_functor() {} 
    __host__ __device__ int operator()(const char t) const { 
     if (t == '\n') return 0; 
     return 1; 
    } 
}; 

void CountPosition1(const char *text, int *pos, int text_size) 
{ 
    thrust::transform(text, text + text_size, pos, text_functor()); 
} 
int main() { 
    char s[4] = {'a', 'a', '\n', 'a'}; 
    int r[4] = {0}; 
    int *k; 
    cudaMalloc((void**) &k, sizeof(int) * 4); 
    CountPosition1(s, k, 4); 
} 

thrust :: transform에서 호스트 반복자와 디바이스 반복자 k를 섞습니다. 세그먼트 화 오류가 발생합니다. CountPosition1에서 인수 k를 r로 변경하면 프로그램이 올바르게됩니다. 추력 함수의 모든 반복기가 동일한 소스 (호스트 또는 두 장치 모두)에 있어야합니까? 아니면이 코드에서 잘못된 점이 있습니까?

Answer 1

예, 모든 반복기는 호스트 컨테이너에 있어야하며 모든 반복기는 장치 컨테이너에 있어야합니다.

알고리즘 디스패치시 thrust will dispatch either the host path or the device path. 모든 반복기는 발송 메소드와 일관성이 있어야합니다.