From 2949e1aa92afb6a494276556b95710c63cba3db3 Mon Sep 17 00:00:00 2001
From: rocking <ChunYu.Lai@amd.com>
Date: Wed, 26 Jul 2023 12:18:56 +0800
Subject: [PATCH] Refine the dimension of host tesnor. This example only
 require 1D (#812)

[ROCm/composable_kernel commit: 016bd428df2554b54007d374955982b1df1369f8]
---
 example/50_put_element/put_element_fp16.cpp | 8 ++++----
 1 file changed, 4 insertions(+), 4 deletions(-)

diff --git a/example/50_put_element/put_element_fp16.cpp b/example/50_put_element/put_element_fp16.cpp
index d4b6831bc4..791747d875 100644
--- a/example/50_put_element/put_element_fp16.cpp
+++ b/example/50_put_element/put_element_fp16.cpp
@@ -34,9 +34,9 @@ int main()
 
     int N = 1024;
 
-    Tensor<XDataType> x(HostTensorDescriptor{N, 1});
-    Tensor<IndexDataType> indices(HostTensorDescriptor{N, 1});
-    Tensor<YDataType> y(HostTensorDescriptor{N, 1});
+    Tensor<XDataType> x(HostTensorDescriptor{N});
+    Tensor<IndexDataType> indices(HostTensorDescriptor{N});
+    Tensor<YDataType> y(HostTensorDescriptor{N});
 
     x.GenerateTensorValue(GeneratorTensor_3<XDataType>{-1.0, 1.0});
     for(int i = 0; i < N; ++i)
@@ -72,7 +72,7 @@ int main()
     bool pass = true;
     if(do_verification)
     {
-        Tensor<YDataType> y_host(HostTensorDescriptor{N, 1});
+        Tensor<YDataType> y_host(HostTensorDescriptor{N});
 
         for(int i = 0; i < N; ++i)
         {