diff --git a/torch_npu/__init__.py b/torch_npu/__init__.py index 755309772ab24cc7188ce734e3daf150c017e2c9..1a3e69126f6b4b34679b266aac323c6696306075 100644 --- a/torch_npu/__init__.py +++ b/torch_npu/__init__.py @@ -255,10 +255,6 @@ def _npu_shutdown(): torch_npu.asd.asd.matmul_check._cleanup() -# register npu shutdown hook on exit -atexit.register(_npu_shutdown) - - # init and register rpc npu backend _rpc_backend_registry()