[Fix] Fix the bug of uninitialized prarmeter (#2345)

* [Fix] Fix the bug of uninitialized prarmeter * refactor code for lint Co-authored-by: budefei <budefei@cambricon.com>

[Fix] Fix the bug of uninitialized prarmeter (#2345)
* [Fix] Fix the bug of uninitialized prarmeter * refactor code for lint Co-authored-by: budefei <budefei@cambricon.com>
04ec054e · bdf · GitHub · be684eeb · 04ec054e
Unverified Commit 04ec054e authored Oct 20, 2022 by bdf Committed by GitHub Oct 20, 2022
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 0 deletions

mmcv/ops/csrc/common/pytorch_mlu_helper.hpp mmcv/ops/csrc/common/pytorch_mlu_helper.hpp +1 -0

No files found.
--- a/mmcv/ops/csrc/common/pytorch_mlu_helper.hpp
+++ b/mmcv/ops/csrc/common/pytorch_mlu_helper.hpp
@@ -27,6 +27,7 @@

 inline int32_t getJobLimitCapability() {
  CNcontext drv_ctx;
+  TORCH_CHECK(CN_SUCCESS == cnCtxGetCurrent(&drv_ctx), "cnCtxGetCurrent fails");
  CNctxConfigParam ctx_conf_param;
  TORCH_CHECK(
      CN_SUCCESS == cnGetCtxConfigParam(drv_ctx, CN_CTX_CONFIG_UNION_LIMIT,