pgjmblvch
/
mindspore2022
forked from huawei/mindspore2022
0
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

duplexpipe ignore python warning 

Signed-off-by: zhoufeng <zhoufeng54@huawei.com>
This commit is contained in:
zhoufeng 2021-03-12 15:03:09 +08:00
parent 98d06d6f7d
commit 9449e848ee
2 changed files with 4 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
mindspore/_extends/remote/kernel_build_server_ascend.py
View File

@ -14,6 +14,7 @@
# ============================================================================
"""kernel build server for ascend"""
import sys
import warnings
from mindspore._extends.remote.kernel_build_server import Messager, get_logger, AkgBuilder
from mindspore._extends.parallel_compile.tbe_compiler.tbe_process import create_tbe_parallel_process, op_select_format
from mindspore._extends.parallel_compile.tbe_compiler.tbe_process import check_supported
@ -140,6 +141,7 @@ class AscendMessager(Messager):
exit()
if __name__ == '__main__':
warnings.simplefilter("ignore")
if len(sys.argv) != 3:
raise Exception('Incorrect argv: {}'.format(sys.argv))
get_logger().debug(f"[TRACE] argv: {str(sys.argv)}")

2
mindspore/_extends/remote/kernel_build_server_gpu.py
View File

@ -15,6 +15,7 @@
"""kernel build server for gpu"""
import os
import sys
import warnings
from mindspore._extends.remote.kernel_build_server import Messager, get_logger, AkgBuilder
from mindspore._extends.parallel_compile.akg_compiler.compiler import run_compiler as akg_compile_single
@ -78,6 +79,7 @@ class GpuMessager(Messager):
exit()
if __name__ == '__main__':
warnings.simplefilter("ignore")
if len(sys.argv) != 3:
raise Exception('Incorrect argv: {}'.format(sys.argv))
get_logger().debug(f"[TRACE] argv: {str(sys.argv)}")