From bdde54c507b10571e210aca18a2079efc70ecbe5 Mon Sep 17 00:00:00 2001
From: zhushuang <zhushuang1110@gmail.com>
Date: Thu, 27 Aug 2026 20:07:10 +0800
Subject: [PATCH] feat(thead): support generated Torch operators

---
 CMakeLists.txt                | 2 +-
 scripts/generate_torch_ops.py | 2 ++
 2 files changed, 3 insertions(+), 1 deletion(-)

diff --git a/CMakeLists.txt b/CMakeLists.txt
index 738846a9e..edc0fd2ea 100644
--- a/CMakeLists.txt
+++ b/CMakeLists.txt
@@ -288,7 +288,7 @@ if(WITH_TORCH OR WITH_LINKED)
     find_library(C10_LIB c10 HINTS ${_torch_lib_dirs} REQUIRED)
     set(TORCH_LIBRARIES ${TORCH_LIB} ${TORCH_CPU_LIB} ${C10_LIB})
 
-    if(WITH_NVIDIA OR WITH_ILUVATAR)
+    if(WITH_NVIDIA OR WITH_ILUVATAR OR WITH_THEAD)
         find_library(TORCH_CUDA_LIB torch_cuda HINTS ${_torch_lib_dirs} REQUIRED)
         find_library(C10_CUDA_LIB c10_cuda HINTS ${_torch_lib_dirs} REQUIRED)
         # `torch_cuda` registers CUDA kernels through static initializers, so
diff --git a/scripts/generate_torch_ops.py b/scripts/generate_torch_ops.py
index 5a6788cf6..6d160c491 100644
--- a/scripts/generate_torch_ops.py
+++ b/scripts/generate_torch_ops.py
@@ -75,6 +75,7 @@
     "kMoore",
     "kIluvatar",
     "kHygon",
+    "kThead",
     "kMars",
 )
 
@@ -84,6 +85,7 @@
     "kMetax",
     "kMoore",
     "kIluvatar",
+    "kThead",
     "kMars",
 )