miopen-wrapper.h 528 B

123456789101112131415161718192021
  1. #pragma once
  2. #include <miopen/miopen.h>
  3. #include <miopen/version.h>
  4. #if MIOPEN_VERSION_MAJOR > 3 || (MIOPEN_VERSION_MAJOR == 3 && MIOPEN_VERSION_MINOR >= 4)
  5. // miopen 3.4 moved find mode from private header to public header
  6. #else
  7. // from miopen_internal.h
  8. extern "C" {
  9. typedef enum
  10. {
  11. miopenConvolutionFindModeNormal = 1, /*!< Normal mode */
  12. } miopenConvolutionFindMode_t;
  13. miopenStatus_t miopenSetConvolutionFindMode(
  14. miopenConvolutionDescriptor_t convDesc,
  15. miopenConvolutionFindMode_t findMode);
  16. }
  17. #endif