Skip to content

GitLab

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
Hide 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() ...@@ -34,9 +34,9 @@ int main()
int N = 1024; int N = 1024;
Tensor<XDataType> x(HostTensorDescriptor{N, 1}); Tensor<XDataType> x(HostTensorDescriptor{N});
Tensor<IndexDataType> indices(HostTensorDescriptor{N, 1}); Tensor<IndexDataType> indices(HostTensorDescriptor{N});
Tensor<YDataType> y(HostTensorDescriptor{N, 1}); Tensor<YDataType> y(HostTensorDescriptor{N});
x.GenerateTensorValue(GeneratorTensor_3<XDataType>{-1.0, 1.0}); x.GenerateTensorValue(GeneratorTensor_3<XDataType>{-1.0, 1.0});
for(int i = 0; i < N; ++i) for(int i = 0; i < N; ++i)
...@@ -72,7 +72,7 @@ int main() ...@@ -72,7 +72,7 @@ int main()
bool pass = true; bool pass = true;
if(do_verification) if(do_verification)
{ {
Tensor<YDataType> y_host(HostTensorDescriptor{N, 1}); Tensor<YDataType> y_host(HostTensorDescriptor{N});
for(int i = 0; i < N; ++i) 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