Add more GPU architectures support (#112)

* Add more GPU architectures support

* Update layout.py

* Optimize performance, Add SM90 support, Add 1D2D SM100 support

* Add fmtlib submodule at commit 553ec11

---------

Co-authored-by: fzyzcjy <5236035+fzyzcjy@users.noreply.github.com>

install.sh

@@ -0,0 +1,13 @@
+# Change current directory into project root
+original_dir=$(pwd)
+script_dir=$(realpath "$(dirname "$0")")
+cd "$script_dir"
+
+# Remove old dist file, build, and install
+rm -rf build dist
+rm -rf *.egg-info
+python setup.py bdist_wheel
+pip install dist/*.whl
+
+# Open users' original directory
+cd "$original_dir"