We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent e38bacb commit 8c1938bCopy full SHA for 8c1938b
deploy/py_infer/src/data_process/postprocess/builder.py
@@ -44,6 +44,7 @@ def get_device_status():
44
def _get_status():
45
nonlocal status
46
try:
47
+ ms.set_context(max_device_memory="0.01GB")
48
status = ms.Tensor([0])[0:].asnumpy()[0]
49
except RuntimeError:
50
status = 1
0 commit comments