build universal pnnx wheel and pnnx entrypoint script (#5331)

.ci/pnnx.yml

@@ -132,7 +132,7 @@ jobs:
       run: |
         export PYTHONUSERBASE=${{ci.workspace}}/torch-${{matrix.torch-version}}
         export PNNX_WHEEL_WITHOUT_BUILD=ON
-        cd tools/pnnx
-        cp build/src/pnnx python/pnnx/
+        cd tools/pnnx/python
+        cp ../build/src/pnnx pnnx/
         python3 setup.py install --user
-        pytest python/tests/
+        pytest tests

.gitignore

@@ -54,10 +54,9 @@ __pycache__
 *.pyd
 *.egg-info/
 python/setup.py
-tools/pnnx/python/setup.py
 
 # Clangd
 .cache/
 
 # Xmake
-.xmake/
+.xmake/

tools/pnnx/python/README.md

@@ -42,7 +42,7 @@ conda install pytorch
 ```
 3. install
 ```bash
-cd /pathto/ncnntools/pnnx
+cd /pathto/ncnntools/pnnx/python
 python setup.py install
 ```
 
@@ -61,8 +61,8 @@ python setup.py install
 
 ## Tests
 ```bash
-cd /pathto/ncnn/tools/pnnx
-pytest python/tests/
+cd /pathto/ncnn/tools/pnnx/python
+pytest tests
 ```
 
 ## Usage

tools/pnnx/python/pnnx/__init__.py

@@ -13,7 +13,10 @@
 # specific language governing permissions and limitations under the License.
 
 import os
+import sys
 import platform
+import subprocess
+
 EXEC_DIR_PATH = os.path.dirname(os.path.abspath(__file__))
 if platform.system() == 'Linux' or platform.system() == "Darwin":
     EXEC_PATH = EXEC_DIR_PATH + "/pnnx"
@@ -31,3 +34,5 @@
 except:
     pass
 
+def pnnx():
+    raise SystemExit(subprocess.call([EXEC_PATH] + sys.argv[1:], close_fds=False))

tools/pnnx/setup.py → tools/pnnx/python/setup.py

@@ -156,10 +156,11 @@ def build_extension(self, ext):
     ],
     license="BSD-3",
     python_requires=">=3.7",
-    packages=find_packages("python"),
+    packages=find_packages(),
     package_data={"pnnx": ["pnnx", "pnnx.exe"]},
-    package_dir={"": "python"},
+    package_dir={"": "."},
     install_requires=requirements,
-    ext_modules=[CMakeExtension("pnnx")],
+    ext_modules=None if PNNX_WHEEL_WITHOUT_BUILD == 'ON' else [CMakeExtension("pnnx", "..")],
     cmdclass={"build_ext": CMakeBuild},
+    entry_points={"console_scripts": ["pnnx=pnnx:pnnx"]},
 )