Browse Source

duplexpipe ignore python warning

Signed-off-by: zhoufeng <zhoufeng54@huawei.com>
tags/v1.2.0-rc1
zhoufeng 4 years ago
parent
commit
9449e848ee
2 changed files with 4 additions and 0 deletions
  1. +2
    -0
      mindspore/_extends/remote/kernel_build_server_ascend.py
  2. +2
    -0
      mindspore/_extends/remote/kernel_build_server_gpu.py

+ 2
- 0
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
- 0
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)}")


Loading…
Cancel
Save