지수는

Question

나는 추력을보고있다 그리고 나는 거의 (그러나 확실히)이 내 대답 질문에 발견 :이 질문에 대해 답이 게시 Finding the maximum element value AND its position using CUDA Thrust

예는 잘 작동하지만, 어떻게 원시 포인터를 사용하여 동일한 작업을 수행 할 수 있습니까? 우리가 내가 올바른 생각하는이 코드를 가정 해 봅시다 (커널 구성을 무시, 그것은 편의상의) :

float* d_A; 
const unsigned int noElems = 10; 
cudaMalloc(&d_A, noElems * sizeof(float)); 
initDeviceVector<<<1, noElems>>>(d_A); 

thrust::device_ptr<float> d_ptr = thrust::device_pointer_cast(d_A);  
thrust::device_vector<float>::iterator iter = 
    thrust::max_element(d_ptr, d_ptr + noElems); 

는 꽤 원시 포인터 ITER 를 사용하여 위치를 추출하는 방법을 알아낼 수 없습니다.

감사합니다.

Answer 1

아마 여러 가지 방법이 있습니다. 코드에서 직접 작업하는 경우 먼저 적절한 장치 포인터로 변환하면 iter의 값을 device_ptr과 비교할 수 있습니다.

$ cat t436.cu 
#include <thrust/device_vector.h> 
#include <thrust/host_vector.h> 
#include <thrust/extrema.h> 
#include <stdio.h> 


__global__ void initDeviceVector(float *data){ 
    int idx = threadIdx.x+blockDim.x*blockIdx.x; 
    data[idx] = idx%7; 
} 

    int main(){ 

    float* d_A; 
    const unsigned int noElems = 10; 
    cudaMalloc(&d_A, noElems * sizeof(float)); 
    initDeviceVector<<<1, noElems>>>(d_A); 

    thrust::device_ptr<float> d_ptr = thrust::device_pointer_cast(d_A); 
    thrust::device_vector<float>::iterator iter = thrust::max_element(d_ptr, d_ptr + noElems); 

    int pos = thrust::device_pointer_cast(&(iter[0])) - d_ptr; 

    printf("pos = %d\n", pos); 
    return 0; 
} 

$ nvcc -arch=sm_20 -o t436 t436.cu 
$ ./t436 
pos = 6 
$