Skip to content

GitLab

"include/ck/utility/get_id.hpp" did not exist on "a0584426ff5b6b8b448c971b97c9b1a4d86ba010"
Unverified Commit 016bd428 authored by rocking's avatar rocking Committed by GitHub
Browse files

Refine the dimension of host tesnor. This example only require 1D (#812)

parent f4ea5601
Show whitespace changes
Inline Side-by-side
Showing with 4 additions and 4 deletions
example/50_put_element/put_element_fp16.cpp
View file @ 016bd428
......@@ -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)
{
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment