AT_DISPATCH_FLOATING_TYPES 매크로는 Pc 텐서의 데이터 타입에 따라 적절한 scalar_t 타입을 선택하고, 그 타입을 사용하여 dist_kernel4 커널을 호출합니다. 매크로의 첫 번째 인자는 텐서의 타입이고, 두 번째 인자는 디스패치 이름(주로 디버깅 용도), 세 번째 인자는 람다 함수입니다.
'Programming > C++' 카테고리의 다른 글
| vscode CMake Debug (0) | 2024.07.10 |
|---|---|
| #pragma unroll, __restrict__ (0) | 2024.07.02 |
| Gstreamer (0) | 2024.04.15 |
| Opencv 설치시 fatal error: va/va.h: No such file or directory 오류 (0) | 2024.03.12 |