CUDA 템플릿 커널 래퍼

Question

나는 시뮬레이션을 쓰기 시작했고보다 객관적인 접근 방식을 사용하기로 결정했다. 따라서 CUDA 커널에서 템플릿 매개 변수를 사용하기로 결정했습니다. 이는 시뮬레이션의 공간 차원을 나타냅니다. 문제는 헤더 파일에 템플릿 함수를 구현하는 데 제한이 있기 때문에 .cpp 소스 파일에서 커널 래퍼를 호출 할 수 있도록 복잡한 접근 방식을 사용해야했습니다.

내 접근 방식은 2 차원과 3 차원에 대한 래퍼 함수를 ​​오버로드하는 것이 었습니다. 그런 다음 초기화 및 커널 리소스 관리 래퍼 클래스에 대한 클래스가 있습니다. 제한의 내가 언급하기 때문에 불행하게도, 내가 템플릿 클래스의 두 멤버를 계속해야 즉

 

struct kernelWrapper{ 
    KernelWrapper(Simulation&lt2> *simulation): 
     d_(2), 
     simulation2d_(simulation) 
    {} 
    KernelWrapper(Simulation&lt3> *simulation): 
     d_(3), 
     simulation3d_(simulation) 
    {} 
    process(void){ //wrapper function for kernel launching 
     switch(d_){ 
     case 2: 
      kernel&lt2><<<..., ...>>>(...); 
      break; 
     } 
     case 3: 
      kernel&lt3><<<..., ...>>>(...); 
      break; 
     } 
     default: 
      break; 
    } 

    int d_; 
    union{ 
     Simulation&lt2> *simulation2d_; 
     Simulation&lt3> *simulation3d_; 
    }; 
    union{ 
     Lattice&lt2> *lattice2d_d; 
     Lattice&lt3> *lattice3d_d; 
    }; 

}; 
 

내가 할 노력하고있어 달성하기 위해 더 나은 방법을 알고 있다면 따라서 궁금

, 즉 템플릿 CUDA 커널에 대한 래퍼를 만듭니다.

업데이트 : 위 게시물을 작성한 후에 알아 낸 해결책을 하나 더 추가하고 싶습니다. C++ faq (포인트 13-15)에 표시된 바와 같이 소스 파일에 템플릿 구현을 넣고 필요한 템플릿을 명시 적으로 인스턴스화 할 수 있습니다. 즉, 2 차원 및 3 차원의 경우에 필요합니다. C++ 11을 사용하면이 단계를 더 진행하고 컴파일/링크 시간을 절약하기 위해 템플릿 정의에 extern 키워드를 도입 할 수 있습니다 (here 설명).

Answer 1

문제 때문에 헤더 파일에 템플릿을 기능을 구현의 제한으로,이다, 나는 에 복잡한 방법을 사용했다가 .CPP 소스 파일에서 호출 커널 래퍼을 유지한다.

템플릿을 작성하는 법적

는 .CPP 코드를 선언

여부 kernelWrapper는 .HPP에 있거나 당신이

template<int d_> 
struct kernelWrapper 
{ 
    KernelWrapper(Simulation<d_> *simulation) : simulation_(simulation) 
    {} 

    process(void) 
    { 
      kernel<d_><<<..., ...>>>(...); 
    } 

    Simulation<d_>* simulation_; 
    Lattice<d_>* lattice2d_; 
}; 

과 같은 코드가 있어야 .CPP 또한 스위치를 사용하지 마십시오/case와 같이 커널을 선택하려면 다음과 같이 사용하십시오 :

int const max_dimension = 4; 

template<int static_dimension> 
void select_kernel(int dynamic_dimension) 
{ 
    if(dynamic_dimension == static_dimension) 
    { 
     call_kernel<static_dimension>(); 
    } 
    select_kernel<static_dimension+1>(dynamic_dimension); 
} 

template<> 
void select_kernel<max_dimension>(int dynamic_dimension) 
{ 
    // error message 
} 

void select_kernel(int dynamic_dimension) 
{ 
    select_kernel<1>(dynamic_dimension); 
} 

그런 선택이 빈번하면, 템플릿을 사용하지 않습니다.