1 #pragma once 2 3 #include <ATen/cuda/PhiloxCudaState.h> 4 #include <ATen/cuda/detail/UnpackRaw.cuh> 5