Skip to content
Projects
Groups
Snippets
Help
Loading...
Sign in / Register
Toggle navigation
N
ngraph
Project
Project
Details
Activity
Cycle Analytics
Repository
Repository
Files
Commits
Branches
Tags
Contributors
Graph
Compare
Charts
Issues
0
Issues
0
List
Board
Labels
Milestones
Merge Requests
0
Merge Requests
0
CI / CD
CI / CD
Pipelines
Jobs
Schedules
Charts
Packages
Packages
Wiki
Wiki
Snippets
Snippets
Members
Members
Collapse sidebar
Close sidebar
Activity
Graph
Charts
Create a new issue
Jobs
Commits
Issue Boards
Open sidebar
submodule
ngraph
Commits
ba9a2a25
Commit
ba9a2a25
authored
7 years ago
by
fenglei.tian
Browse files
Options
Browse Files
Download
Email Patches
Plain Diff
clang format
parent
d0ab30a2
Hide whitespace changes
Inline
Side-by-side
Showing
2 changed files
with
10 additions
and
8 deletions
+10
-8
gpu_emitter.cpp
src/ngraph/runtime/gpu/gpu_emitter.cpp
+6
-6
backend_performance.cpp
test/backend_performance.cpp
+4
-2
No files found.
src/ngraph/runtime/gpu/gpu_emitter.cpp
View file @
ba9a2a25
...
...
@@ -193,12 +193,12 @@ void runtime::gpu::GPU_Emitter::EmitDot(codegen::CodeWriter& writer,
else
if
((
arg0_shape
.
size
()
==
2
)
&&
(
arg1_shape
.
size
()
==
2
))
{
// GEMM Call
if
(
arg0_shape
[
0
]
!=
out
[
0
].
get_shape
()[
0
]
||
// m
arg1_shape
[
1
]
!=
out
[
0
].
get_shape
()[
1
]
||
// n
arg0_shape
[
1
]
!=
arg1_shape
[
0
])
// k
{
throw
std
::
runtime_error
(
"input and output shape is not correct for dot;"
);
}
if
(
arg0_shape
[
0
]
!=
out
[
0
].
get_shape
()[
0
]
||
// m
arg1_shape
[
1
]
!=
out
[
0
].
get_shape
()[
1
]
||
// n
arg0_shape
[
1
]
!=
arg1_shape
[
0
])
// k
{
throw
std
::
runtime_error
(
"input and output shape is not correct for dot;"
);
}
writer
<<
"{ // "
<<
n
->
get_name
()
<<
"
\n
"
;
writer
.
indent
++
;
writer
<<
"static const float alpha = 1.0;
\n
"
;
...
...
This diff is collapsed.
Click to expand it.
test/backend_performance.cpp
View file @
ba9a2a25
...
...
@@ -162,13 +162,15 @@ TEST(benchmark, mxnet_seq2seq_backward)
TEST
(
benchmark
,
mxnet_sockeye_seq2seq_forward
)
{
const
string
json_path
=
file_util
::
path_join
(
SERIALIZED_ZOO
,
"mxnet/Sockeye_Seq2Seq_forward.json"
);
const
string
json_path
=
file_util
::
path_join
(
SERIALIZED_ZOO
,
"mxnet/Sockeye_Seq2Seq_forward.json"
);
run_benchmark
(
json_path
,
"CPU"
,
10
);
}
TEST
(
benchmark
,
mxnet_sockeye_seq2seq_backward
)
{
const
string
json_path
=
file_util
::
path_join
(
SERIALIZED_ZOO
,
"mxnet/Sockeye_Seq2Seq_backward.json"
);
const
string
json_path
=
file_util
::
path_join
(
SERIALIZED_ZOO
,
"mxnet/Sockeye_Seq2Seq_backward.json"
);
run_benchmark
(
json_path
,
"CPU"
,
10
);
}
...
...
This diff is collapsed.
Click to expand it.
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