정적 추력 사용자 지정 할당 기? 설정을위한

Question

그냥 몇 가지 사실 :

• 추력은 그것의 모든 조작에 적절한 작동하지 않습니다.
• 사용자 지정 할당자를 thrust::device_vectors에 제공 할 수 있습니다.

나는 thrust::system과 thrust::system::cuda을보고 정적 시스템 할당 자처럼 보이는 것을 찾지 못했습니다. 내 말은, 추력이 내부적으로 사용하는 할당자를 out-of-place 알고리즘에 여분의 메모리를 할당하는 대신 교체 할 수는 없다는 것입니다.

또한 할당자가 작업 메모리에 주어진 thrust::device_vectors에 대해 할당자를 사용한다고 생각하기가 어렵습니다.

질문 : thrust에는 내부 할당자를 사용자 정의 된 것으로 바꾸는 방법이 있습니까?

관련 질문 :

implies that thrust operates out-of-place

example of custom thrust allocator

Answer 1

추력의 custom_temporary_allocation 예는 추력 알고리즘에 의해 내부적으로 사용하는 임시 저장을 위해 자신 만의 할당을 구축하는 방법을 보여줍니다. 이 예제는 캐싱 체계를 사용하여 할당을 수행하지만 원칙적으로 원하는 전략을 사용할 수 있습니다.

기본적으로 할당을 사용자 지정하기 위해 CUDA 백엔드에서 파생 된 사용자 지정 백엔드를 만드는 것이 아이디어입니다. 그런 다음 사용자 정의 할당자를 사용하여 알고리즘을 사용하려면 알고리즘을 호출 할 때 사용자 정의 백엔드에서 Thrust를 가리 킵니다.

이 기능에는 1.6 이상의 트러스트가 필요합니다.