| 19 | } |
| 20 | |
| 21 | megcoreStatus_t megcore::getROCMContext( |
| 22 | megcoreComputingHandle_t handle, ROCMContext* ctx) { |
| 23 | auto&& H = handle; |
| 24 | megdnn_assert(H); |
| 25 | megcoreDeviceHandle_t dev_handle = H->content->dev_handle(); |
| 26 | megcorePlatform_t platform; |
| 27 | megcoreGetPlatform(dev_handle, &platform); |
| 28 | megdnn_assert(platform == megcorePlatformROCM); |
| 29 | auto context = static_cast<megcore::rocm::ROCMComputingContext*>(H->content.get()); |
| 30 | *ctx = context->context(); |
| 31 | return megcoreSuccess; |
| 32 | } |
| 33 | |
| 34 | std::atomic_bool megcore::ROCMContext::sm_miopen_algo_search{false}; |
| 35 | megcoreStatus_t megcore::enableMIOpenAlgoSearch(bool enable_algo_search) { |
nothing calls this directly
no test coverage detected