fp16.h 395 B

12345678910111213141516
  1. #if !defined(TORCH_STABLE_ONLY) && !defined(TORCH_TARGET_VERSION)
  2. #pragma once
  3. #ifndef FP16_H
  4. #define FP16_H
  5. #include <fp16/fp16.h>
  6. #if defined(PSIMD_H)
  7. #include <fp16/psimd.h>
  8. #endif
  9. #endif /* FP16_H */
  10. #else
  11. #error "This file should not be included when either TORCH_STABLE_ONLY or TORCH_TARGET_VERSION is defined."
  12. #endif // !defined(TORCH_STABLE_ONLY) && !defined(TORCH_TARGET_VERSION)