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>

develop.sh

@@ -0,0 +1,25 @@
+# Change current directory into project root
+original_dir=$(pwd)
+script_dir=$(realpath "$(dirname "$0")")
+cd "$script_dir"
+
+# Link CUTLASS includes
+ln -sf $script_dir/third-party/cutlass/include/cutlass deep_gemm/include
+ln -sf $script_dir/third-party/cutlass/include/cute deep_gemm/include
+
+# Remove old dist file, build, and build
+rm -rf build dist
+rm -rf *.egg-info
+python setup.py build
+
+# Find the .so file in build directory and create symlink in current directory
+so_file=$(find build -name "*.so" -type f | head -n 1)
+if [ -n "$so_file" ]; then
+    ln -sf "$so_file" .
+else
+    echo "Error: No SO file found in build directory" >&2
+    exit 1
+fi
+
+# Open users' original directory
+cd "$original_dir"