fix headers (#2321)

[ROCm/composable_kernel commit: 1ac5eeaea9]
2026-05-14 02:02:46 +00:00 · 2025-06-10 07:26:32 -07:00
parent e2c603e173
commit c3ee65fd0f
2 changed files with 2 additions and 5 deletions
--- a/library/include/ck/library/tensor_operation_instance/add_device_operation_instance.hpp
+++ b/library/include/ck/library/tensor_operation_instance/add_device_operation_instance.hpp
@@ -5,6 +5,7 @@

 #include <vector>
 #include <type_traits>
+#include <memory>

 #include "ck/utility/functional2.hpp"

--- a/test/scatter_gather/scatter_gather.cpp
+++ b/test/scatter_gather/scatter_gather.cpp
@@ -1,13 +1,9 @@
 // SPDX-License-Identifier: MIT
 // Copyright (c) 2018-2024, Advanced Micro Devices, Inc. All rights reserved.

-#include <vector>
-#include <iostream>
-#include <numeric>
 #include <cassert>
 #include <cstdlib>
-#include <iostream>
-#include <time.h>
+#include <ctime>
 #include <unordered_set>

 #include "ck_tile/core.hpp"