From 0ef3bc3ab55441c1e8a006f5c6fd654f5d21407a Mon Sep 17 00:00:00 2001
From: wooway777 <wooway777@gmail.com>
Date: Thu, 2 Jul 2026 16:24:20 +0800
Subject: [PATCH] feat: allow causal conv1d on hygon

---
 src/infiniop/ops/causal_conv1d/operator.cc | 16 +++++++++++++---
 1 file changed, 13 insertions(+), 3 deletions(-)

diff --git a/src/infiniop/ops/causal_conv1d/operator.cc b/src/infiniop/ops/causal_conv1d/operator.cc
index 1e17605c9..eacee4c1a 100644
--- a/src/infiniop/ops/causal_conv1d/operator.cc
+++ b/src/infiniop/ops/causal_conv1d/operator.cc
@@ -1,10 +1,8 @@
-// infiniop/ops/causal_conv1d/operator.cc
-
 #include "../../operator.h"
 #include "../../handle.h"
 #include "infiniop/ops/causal_conv1d.h"
 
-#if defined(ENABLE_NVIDIA_API)
+#if defined(ENABLE_NVIDIA_API) || defined(ENABLE_HYGON_API)
 #include "nvidia/causal_conv1d_nvidia.cuh"
 #endif
 
@@ -33,6 +31,9 @@ __INFINI_C infiniStatus_t infiniopCreateCausalConv1dDescriptor(
     switch (handle->device) {
 #ifdef ENABLE_NVIDIA_API
         CREATE(INFINI_DEVICE_NVIDIA, nvidia)
+#endif
+#ifdef ENABLE_HYGON_API
+        CREATE(INFINI_DEVICE_HYGON, nvidia)
 #endif
     default:
         return INFINI_STATUS_DEVICE_TYPE_NOT_SUPPORTED;
@@ -50,6 +51,9 @@ __INFINI_C infiniStatus_t infiniopGetCausalConv1dWorkspaceSize(
     switch (desc->device_type) {
 #ifdef ENABLE_NVIDIA_API
         GET(INFINI_DEVICE_NVIDIA, nvidia)
+#endif
+#ifdef ENABLE_HYGON_API
+        GET(INFINI_DEVICE_HYGON, nvidia)
 #endif
     default:
         return INFINI_STATUS_DEVICE_TYPE_NOT_SUPPORTED;
@@ -81,6 +85,9 @@ __INFINI_C infiniStatus_t infiniopCausalConv1d(
     switch (desc->device_type) {
 #ifdef ENABLE_NVIDIA_API
         CALCULATE(INFINI_DEVICE_NVIDIA, nvidia)
+#endif
+#ifdef ENABLE_HYGON_API
+        CALCULATE(INFINI_DEVICE_HYGON, nvidia)
 #endif
     default:
         return INFINI_STATUS_DEVICE_TYPE_NOT_SUPPORTED;
@@ -98,6 +105,9 @@ __INFINI_C infiniStatus_t infiniopDestroyCausalConv1dDescriptor(
     switch (desc->device_type) {
 #ifdef ENABLE_NVIDIA_API
         DESTROY(INFINI_DEVICE_NVIDIA, nvidia)
+#endif
+#ifdef ENABLE_HYGON_API
+        DESTROY(INFINI_DEVICE_HYGON, nvidia)
 #endif
     default:
         return INFINI_STATUS_DEVICE_TYPE_NOT_SUPPORTED;