move mindnlp.core to mindtorch (#2168)

Author: lvyufeng

mindnlp/__init__.py

@@ -17,44 +17,13 @@
 MindNLP library.
 """
 import os
-import platform
 
 # huggingface env
 if os.environ.get('HF_ENDPOINT', None) is None:
     os.environ["HF_ENDPOINT"] = 'https://hf-mirror.com'
 
-# for huawei cloud modelarts
-if 'RANK_TABLE_FILE' in os.environ:
-    del os.environ['RANK_TABLE_FILE']
-
-import mindspore
-from mindspore._c_expression import MSContext # pylint: disable=no-name-in-module, import-error
-try:
-    from mindspore._c_expression import disable_multi_thread
-except:
-    disable_multi_thread = None
-
-if os.environ.get('DEVICE_TARGET', None) is not None:
-    mindspore.set_device(os.environ.get('DEVICE_TARGET'))
-
-# for different ascend devices
-if platform.system().lower() == 'linux' and mindspore.get_context('device_target') == 'Ascend':
-    SOC = MSContext.get_instance().get_ascend_soc_version()
-    # enable vmm since only vmm can release device memory when del tensor.
-    if SOC != 'ascend310b':
-        os.environ["MS_ALLOC_CONF"] = 'enable_vmm:True,vmm_align_size:2MB'
-
-    if SOC in ('ascend910', 'ascend310b'):
-        # context.set_context(ascend_config={"precision_mode": "allow_mix_precision"})
-        mindspore.device_context.ascend.op_precision.precision_mode('allow_mix_precision')
-    if SOC == 'ascend310b' and disable_multi_thread is not None:
-        disable_multi_thread()
-
 # set mindnlp.core to torch
-from .utils.torch_proxy import initialize_torch_proxy, setup_metadata_patch
-initialize_torch_proxy()
-setup_metadata_patch()
-
+import mindtorch
 from .utils.safetensors_patch import setup_safetensors_patch
 setup_safetensors_patch()