support tensor parallel #2622

Job	Run time
build-and-test-cpp-x86_64 (ubuntu-20.04, mkl)	4m 30s
build-and-push-docker-images	14m 4s
build-and-test-cpp-aarch64 (openblas)	4m 5s
build-and-test-cpp-x86_64 (ubuntu-20.04, dnnl)	4m 31s
build-python-wheels (ubuntu-20.04, auto64)	2m 8s
build-and-test-cpp-x86_64 (macos-11, mkl)	4m 28s
build-python-wheels (macos-11, auto64)	2m 8s
build-and-test-cpp-x86_64 (macos-11, dnnl)	4m 11s
check-python-style	7s
build-python-wheels (windows-2019, auto64)	2m 22s
build-python-wheels (ubuntu-20.04, aarch64)	2m 11s
build-python-wheels (macos-11, arm64)	1m 50s
build-and-deploy-docs	0s
test-python-wheels	0s
publish-python-wheels-on-pypi	0s
	46m 35s

Provide feedback