PhiloxUtils.cuh 349 B

123456789
  1. #if !defined(TORCH_STABLE_ONLY) && !defined(TORCH_TARGET_VERSION)
  2. #pragma once
  3. #include <ATen/cuda/PhiloxCudaState.h>
  4. #include <ATen/cuda/detail/UnpackRaw.cuh>
  5. #else
  6. #error "This file should not be included when either TORCH_STABLE_ONLY or TORCH_TARGET_VERSION is defined."
  7. #endif // !defined(TORCH_STABLE_ONLY) && !defined(TORCH_TARGET_VERSION)