You can not select more than 25 topics Topics must start with a chinese character,a letter or number, can include dashes ('-') and can be up to 35 characters long.

hip_header.h 795 B

123456789101112131415161718192021222324
  1. #pragma once
  2. /**
  3. * \remarks The files in the subdirectory include/hip are copied from HIP
  4. * headers provided by ROCm-Developer-Tools/HIP, which can be found from
  5. * https://github.com/ROCm-Developer-Tools/HIP. These files are included to make
  6. * the MegDNN can be compiled with both CUDA and ROCm backends, and the both
  7. * backends share the same code.
  8. */
  9. #pragma GCC diagnostic push
  10. #pragma GCC diagnostic ignored "-Wunused-parameter"
  11. #pragma GCC diagnostic ignored "-Wdeprecated-declarations"
  12. #pragma GCC diagnostic ignored "-Wsign-compare"
  13. #include <hip/hip_fp16.h>
  14. #include <hip/hip_runtime.h>
  15. #include <hip/hip_runtime_api.h>
  16. #pragma GCC diagnostic pop
  17. #if !defined(__HIP_PLATFORM_HCC__)
  18. #error "platform macro __HIP_PLATFORM_HCC__ must be defined"
  19. #endif
  20. // vim: syntax=cpp.doxygen