추력 : 알고리즘 호출이 실행되는 위치를 직접 제어하는 ​​방법은 무엇입니까?

Question

다음 코드에는 CPU 또는 GPU에서 실행될 수있는 정보가 없습니다. 나는 "reduce"연산이 어디에서 실행되는지 궁금합니다.

#include <thrust/iterator/counting_iterator.h> 
... 
// create iterators 
thrust::counting_iterator<int> first(10); 
thrust::counting_iterator<int> last = first + 3; 

first[0] // returns 10 
first[1] // returns 11 
first[100] // returns 110 

// sum of [first, last) 
thrust::reduce(first, last); // returns 33 (i.e. 10 + 11 + 12) 

또한

는

데이터 추력 :: host_vector 정의하더라도

thrust::transform_reduce(
    thrust::counting_iterator<unsigned int>(0), 
    thrust::counting_iterator<unsigned int>(N), 
    MyOperation(data), 0 ,thrust::plus<unsigned int>()) 

이 기능은 GPU (파일 이름의 .cpp 끝나는 때문에 컴파일러는 관련 에러를 범)에서 실행하려고. CPU에서 코드를 실행하려면 어떻게해야합니까? 또는 동일한 작업을 수행하는 또 다른 방법을 찾아야합니까? counting_iterator를 사용하지 않습니까?

Answer 1

기본적으로 이와 같은 알고리즘 호출은 기기 백엔드 (즉, 사례의 GPU)에서 실행됩니다.

#include <thrust/execution_policy.h> 

... 

thrust::reduce(thrust::host, first, last); 

... 

thrust::transform_reduce(thrust::host, 
         first, 
         last, 
         MyOperation(data), 
         0, 
         thrust::plus<unsigned int>()); 

는 1.6 추력 사용하는 경우 : 당신은 추력 1.7 이상을 사용하는 경우

는 호스트 (즉, CPU)에서 실행하는 알고리즘 호출을 강제로 thrust::host 실행 정책을 사용 , 기존 반복자 드래그하는 retag하여 호스트로 호출 목표를 재설정 할 수 있습니다 : 당신은 추력 전 1.6의 이전 버전을 사용하는 경우

#include <thrust/iterator/retag.h> 

... 

thrust::reduce(thrust::retag<thrust::host_system_tag>(first), 
       thrust::retag<thrust::host_system_tag>(last)); 

... 

thrust::transform_reduce(thrust::retag<thrust::host_system_tag>(first), 
         thrust::retag<thrust::host_system_tag>(last), 
         MyOperation(data), 
         0, 
         thrust::plus<unsigned int>()); 

를 템플릿으로 counting_iterator에 host_space_tag를 전달해야 매개 변수 :

thrust::reduce(thrust::counting_iterator<unsigned int, thrust::host_space_tag>(0), 
       thrust::counting_iterator<unsigned int, thrust::host_space_tag>(N));