From 3303c0bed036c9fe9cd208bf0974e4da9f98ed20 Mon Sep 17 00:00:00 2001
From: Zhiyuan Chen <136175529+Jassicia@users.noreply.github.com>
Date: Mon, 8 Jun 2026 20:20:49 +0800
Subject: [PATCH] Add files via upload

---
 TopK.mlu | 2 +-
 config   | 3 ---
 2 files changed, 1 insertion(+), 4 deletions(-)

diff --git a/TopK.mlu b/TopK.mlu
index 6f2d000..76f098c 100644
--- a/TopK.mlu
+++ b/TopK.mlu
@@ -68,7 +68,7 @@ std::vector<torch::Tensor> bang_func(torch::Tensor x, int k, int dim) {
     auto indices = torch::empty({batch, k}, x.options().dtype(torch::kInt64));
 
     cnrtQueue_t queue = torch_mlu::getCurMLUStream();
-    cnrtDim3_t task_dim = {batch, 1, 1};
+    cnrtDim3_t task_dim = {static_cast<unsigned int>(batch), 1, 1};
     cnrtFunctionType_t ktype = cnrtFuncTypeUnion1;
 
     topk_dim1_kernel<<<task_dim, ktype, queue>>>(
diff --git a/config b/config
index e038ed4..920a6ea 100644
--- a/config
+++ b/config
@@ -1,4 +1 @@
-070
 075
-051
-