doc

example/ck_tile/22_documentation_examples/buffer_examples.cpp

@@ -0,0 +1,47 @@
+// SPDX-License-Identifier: MIT
+// Copyright (c) 2024-2025, Advanced Micro Devices, Inc. All rights reserved.
+
+#include "ck_tile/host.hpp"
+#include <cstring>
+#include "ck_tile/ops/documentation_examples/kernel/example_buffer_kernel.hpp"
+
+template <typename DataType>
+bool run()
+{
+    using XDataType       = DataType;
+
+    auto problem_shape = ck_tile::make_tuple(10);
+
+    ck_tile::HostTensor<XDataType> x_host({10}, {1});
+
+    for(int i=0; i<10; i++)
+    {
+        x_host.data()[i] = static_cast<XDataType>(i);
+    }
+
+    ck_tile::DeviceMem x_buf(x_host.get_element_space_size_in_bytes());
+
+    x_buf.ToDevice(x_host.data());
+
+    constexpr ck_tile::index_t kBlockSize  = 1;
+    constexpr ck_tile::index_t kBlockPerCu = 1;    
+    constexpr ck_tile::index_t kGridSize = 1;
+
+    using Kernel = ck_tile::Docs<XDataType>;
+
+    launch_kernel(ck_tile::stream_config{nullptr, true, 0, 0, 1},
+                                   ck_tile::make_kernel<kBlockSize, kBlockPerCu>(
+                                       Kernel{},
+                                       1,
+                                       kBlockSize,
+                                       0,
+                                       static_cast<XDataType*>(x_buf.GetDeviceBuffer()),
+                                       problem_shape));
+
+    return true;
+}
+
+int main()
+{
+    return run<int>() ? 0 : -2;
+}