Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
gaoqiong
composable_kernel
Commits
7d5d9b8f
".github/vscode:/vscode.git/clone" did not exist on "51679bbda84df6e68d4cef8f59abf8f962bf59e0"
Commit
7d5d9b8f
authored
Sep 08, 2022
by
Po-Yen, Chen
Browse files
Fill tensor values start from 1
parent
33aa4d45
Changes
1
Show whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
1 addition
and
1 deletion
+1
-1
example/36_permute/run_permute_example.inc
example/36_permute/run_permute_example.inc
+1
-1
No files found.
example/36_permute/run_permute_example.inc
View file @
7d5d9b8f
...
...
@@ -14,7 +14,7 @@ bool run_permute(const ExecutionConfig& config, const Problem& problem)
Tensor
<
ADataType
>
a
(
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
b_device_buf
(
sizeof
(
BDataType
)
*
b
.
mDesc
.
GetElementSpaceSize
());
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment