diff --git a/python/setup_cutlass.py b/python/setup_cutlass.py
index d057de16..150d92a6 100644
--- a/python/setup_cutlass.py
+++ b/python/setup_cutlass.py
@@ -51,7 +51,7 @@ setup_pycute.perform_setup()
 
 setup(
     name='cutlass_cppgen',
-    version='4.3.5',
+    version='4.4.0',
     description='CUTLASS Pythonic Interface',
     package_dir={'': '.'},
     packages=[