From 481a5acf1f33a23cf918f4bca14a63fcfdd1ddf8 Mon Sep 17 00:00:00 2001
From: Xu Zhao <i@xuzhao.net>
Date: Thu, 19 Dec 2024 11:25:20 -0500
Subject: [PATCH] Upgrade tk

---
 install.py                | 2 +-
 submodules/ThunderKittens | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/install.py b/install.py
index 9658a14..92279a1 100644
--- a/install.py
+++ b/install.py
@@ -139,7 +139,7 @@ def setup_hip(args: argparse.Namespace):
     if args.jax or args.all:
         logger.info("[tritonbench] installing jax...")
         install_jax()
-    if args.tk or args.all:
+    if args.tk:
         logger.info("[tritonbench] installing thunderkittens...")
         from tools.tk.install import install_tk
 
diff --git a/submodules/ThunderKittens b/submodules/ThunderKittens
index 5d7107a..cdfce88 160000
--- a/submodules/ThunderKittens
+++ b/submodules/ThunderKittens
@@ -1 +1 @@
-Subproject commit 5d7107a13b016811da38e781d7fc4f74140b2d03
+Subproject commit cdfce886b2660a0345a20bc5c7d52efd0db95fc0