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.
|
- #pragma once
-
- /**
- * \remarks The files in the subdirectory include/hip are copied from HIP
- * headers provided by ROCm-Developer-Tools/HIP, which can be found from
- * https://github.com/ROCm-Developer-Tools/HIP. These files are included to make
- * the MegDNN can be compiled with both CUDA and ROCm backends, and the both
- * backends share the same code.
- */
-
- #pragma GCC diagnostic push
- #pragma GCC diagnostic ignored "-Wunused-parameter"
- #pragma GCC diagnostic ignored "-Wdeprecated-declarations"
- #pragma GCC diagnostic ignored "-Wsign-compare"
- #include <hip/hip_fp16.h>
- #include <hip/hip_runtime.h>
- #include <hip/hip_runtime_api.h>
- #pragma GCC diagnostic pop
-
- #if !defined(__HIP_PLATFORM_HCC__)
- #error "platform macro __HIP_PLATFORM_HCC__ must be defined"
- #endif
-
- // vim: syntax=cpp.doxygen
|