mirrors
/
mmcv
mirror of https://github.com/open-mmlab/mmcv.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

[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>
pull/2311/head^2
bdf 2022-10-20 20:12:15 +08:00 committed by GitHub
parent be684eeb4c
commit 04ec054e6b
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
mmcv/ops/csrc/common/pytorch_mlu_helper.hpp
View File

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