Support both cpu & cuda backends in Dockerfile (#101) #87

	# This workflow will install Python dependencies, run tests and lint with a variety of Python versions
	# For more information see: https://docs.github.com/en/actions/automating-builds-and-tests/building-and-testing-python

	name: Python package

	on:
	push:
	branches: [ "main" ]
	pull_request:
	branches: [ "main" ]

	jobs:
	build:

	runs-on: ${{ matrix.os }}
	strategy:
	fail-fast: false
	matrix:
	os: [ubuntu-latest, macos-latest]
	python-version: ["3.8", "3.9", "3.10", "3.11"]

	steps:
	- uses: actions/checkout@v3
	with:
	submodules: true
	- name: Set up Python ${{ matrix.python-version }}
	uses: actions/setup-python@v4
	with:
	python-version: ${{ matrix.python-version }}
	- name: Install dependencies
	run: \|
	python -m pip install --upgrade pip
	python -m pip install pytest build
	python -m build --sdist
	pip install dist/*.tar.gz -v
	- name: Lint with black
	uses: psf/black@stable
	with:
	options: "--check --verbose --line-length 120"
	src: "chatglm_cpp examples tests setup.py"
	- name: Test with pytest
	run: \|
	cd tests
	pytest test_chatglm_cpp.py

	build-windows:

	runs-on: windows-latest

	steps:
	- uses: actions/checkout@v3
	with:
	submodules: true
	- name: Set up Python 3.8
	uses: actions/setup-python@v4
	with:
	python-version: "3.8"
	- name: Install dependencies
	run: \|
	python -m pip install --upgrade pip
	python -m pip install pytest
	pip install . -v
	- name: Test with pytest
	run: \|
	cd tests
	pytest test_chatglm_cpp.py

Provide feedback