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
949de23b
Commit
949de23b
authored
Aug 29, 2023
by
Astha Rai
Browse files
example with 5D input tensor and 2d kernel - not working: issues with arguments
parent
b472cdf6
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
11 additions
and
5 deletions
+11
-5
example/44_elementwise_permute/elementwise_permute_4D_fp16_2d.cpp
...44_elementwise_permute/elementwise_permute_4D_fp16_2d.cpp
+11
-5
No files found.
example/44_elementwise_permute/elementwise_permute_4D_fp16_2d.cpp
View file @
949de23b
...
...
@@ -48,11 +48,14 @@ int main()
bool
do_verification
=
true
;
bool
time_kernel
=
true
;
const
int
N
=
120
;
const
int
C
=
128
;
const
int
H
=
32
;
const
int
W
=
1024
;
//const int N = 120;
//const int C = 128;
//const int H = 32;
//const int W = 1024;
const
int
N
=
16
;
const
int
C
=
8
;
const
int
H
=
8
;
const
int
W
=
8
;
/**const int N = 120;
const int H = 32;
const int W = 64;
...
...
@@ -107,6 +110,8 @@ int main()
float
gb_per_sec
=
num_btype
/
1.E6
/
ave_time
;
LogRangeAsType
<
float
>
(
std
::
cout
<<
"A : "
,
a
.
mData
,
","
)
<<
std
::
endl
;
//LogRangeAsType<float>(std::cout << "B : ", b.mData, ",") << std::endl;
std
::
cout
<<
"Perf: "
<<
ave_time
<<
" ms, "
<<
tflops
<<
" TFlops, "
<<
gb_per_sec
<<
" GB/s"
<<
std
::
endl
;
...
...
@@ -120,6 +125,7 @@ int main()
Tensor
<
BDataType
>
host_b
(
nhwc
);
host_elementwise4D
<
Tensor
<
ADataType
>
,
Tensor
<
BDataType
>
,
PassThrough
>
(
host_b
,
a
,
nchw
,
PassThrough
{});
//LogRangeAsType<float>(std::cout << "Host_b : ", host_b.mData, ",") << std::endl;
// LogRangeAsType<float>(std::cout << "Host b : ", host_b.mData, ",") << std::endl;
pass
&=
...
...
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