Programming/C++

AT_DISPATCH_FLOATING_TYPES, AT_ASSERTM

zz0622 2024. 7. 2. 09:12

AT_DISPATCH_FLOATING_TYPES 매크로는 Pc 텐서의 데이터 타입에 따라 적절한 scalar_t 타입을 선택하고, 그 타입을 사용하여 dist_kernel4 커널을 호출합니다. 매크로의 첫 번째 인자는 텐서의 타입이고, 두 번째 인자는 디스패치 이름(주로 디버깅 용도), 세 번째 인자는 람다 함수입니다.