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
MIGraphX
Commits
aaf8b162
"include/vscode:/vscode.git/clone" did not exist on "85bddfbcc64daeca26727b179779ac727d26ddf7"
Commit
aaf8b162
authored
May 13, 2019
by
Shucai Xiao
Browse files
change the type of alpha and beta from int8 to int32 for quant_dot
parent
f0f562b2
Changes
5
Hide whitespace changes
Inline
Side-by-side
Showing
5 changed files
with
6 additions
and
6 deletions
+6
-6
src/include/migraphx/op/quant_dot.hpp
src/include/migraphx/op/quant_dot.hpp
+2
-2
src/targets/cpu/gemm.cpp
src/targets/cpu/gemm.cpp
+1
-1
src/targets/cpu/include/migraphx/cpu/gemm.hpp
src/targets/cpu/include/migraphx/cpu/gemm.hpp
+1
-1
src/targets/cpu/lowering.cpp
src/targets/cpu/lowering.cpp
+1
-1
src/targets/gpu/quant_gemm.cpp
src/targets/gpu/quant_gemm.cpp
+1
-1
No files found.
src/include/migraphx/op/quant_dot.hpp
View file @
aaf8b162
...
...
@@ -18,8 +18,8 @@ namespace op {
struct
quant_dot
{
int
8
_t
alpha
=
1
;
int
8
_t
beta
=
1
;
int
32
_t
alpha
=
1
;
int
32
_t
beta
=
1
;
template
<
class
Self
,
class
F
>
static
auto
reflect
(
Self
&
self
,
F
f
)
...
...
src/targets/cpu/gemm.cpp
View file @
aaf8b162
...
...
@@ -119,7 +119,7 @@ void migemm(
}
void
migemm
(
const
argument
&
c_arg
,
const
argument
&
a_arg
,
const
argument
&
b_arg
,
int
8
_t
alpha
,
int
8
_t
beta
)
const
argument
&
c_arg
,
const
argument
&
a_arg
,
const
argument
&
b_arg
,
int
32
_t
alpha
,
int
32
_t
beta
)
{
migemm_tpl
(
c_arg
,
a_arg
,
b_arg
,
alpha
,
beta
);
}
...
...
src/targets/cpu/include/migraphx/cpu/gemm.hpp
View file @
aaf8b162
...
...
@@ -11,7 +11,7 @@ namespace cpu {
void
migemm
(
const
argument
&
c_arg
,
const
argument
&
a_arg
,
const
argument
&
b_arg
,
float
alpha
,
float
beta
);
void
migemm
(
const
argument
&
c_arg
,
const
argument
&
a_arg
,
const
argument
&
b_arg
,
int
8
_t
alpha
,
int
8
_t
beta
);
const
argument
&
c_arg
,
const
argument
&
a_arg
,
const
argument
&
b_arg
,
int
32
_t
alpha
,
int
32
_t
beta
);
}
// namespace cpu
}
// namespace MIGRAPHX_INLINE_NS
...
...
src/targets/cpu/lowering.cpp
View file @
aaf8b162
...
...
@@ -563,7 +563,7 @@ struct cpu_quant_gemm
}
// 2 input arguments
int
8
_t
beta
=
0
;
int
32
_t
beta
=
0
;
migemm
(
result
,
arg_0
,
arg_1
,
op
.
alpha
,
beta
);
return
result
;
...
...
src/targets/gpu/quant_gemm.cpp
View file @
aaf8b162
...
...
@@ -94,7 +94,7 @@ argument miopen_quant_gemm::compute(context& ctx,
}
bool
is_3inputs
=
(
args
.
size
()
==
4
);
int
8
_t
beta
=
0
;
int
32
_t
beta
=
0
;
if
(
is_3inputs
)
{
beta
=
op
.
beta
;
...
...
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