Skip to content

GitLab

Commit 7d5d9b8f authored by Po-Yen, Chen's avatar Po-Yen, Chen
Browse files

Fill tensor values start from 1

parent 33aa4d45
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
example/36_permute/run_permute_example.inc
View file @ 7d5d9b8f
...@@ -14,7 +14,7 @@ bool run_permute(const ExecutionConfig& config, const Problem& problem) ...@@ -14,7 +14,7 @@ bool run_permute(const ExecutionConfig& config, const Problem& problem)
Tensor<ADataType> a(shape); Tensor<ADataType> a(shape);
Tensor<BDataType> b(transposed_shape); Tensor<BDataType> b(transposed_shape);
std::iota(begin(a.mData), end(a.mData), 0); std::iota(begin(a.mData), end(a.mData), 1);
DeviceMem a_device_buf(sizeof(ADataType) * a.mDesc.GetElementSpaceSize()); DeviceMem a_device_buf(sizeof(ADataType) * a.mDesc.GetElementSpaceSize());
DeviceMem b_device_buf(sizeof(BDataType) * b.mDesc.GetElementSpaceSize()); DeviceMem b_device_buf(sizeof(BDataType) * b.mDesc.GetElementSpaceSize());
......
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