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
3443835c
Commit
3443835c
authored
Mar 09, 2021
by
Chao Liu
Browse files
bug fix: add missing data-type in inner_product_with_conversion
parent
59968d8d
Changes
3
Hide whitespace changes
Inline
Side-by-side
Showing
3 changed files
with
8 additions
and
4 deletions
+8
-4
composable_kernel/include/utility/config.amd.hpp.in
composable_kernel/include/utility/config.amd.hpp.in
+1
-1
composable_kernel/include/utility/float_type.amd.hpp.in
composable_kernel/include/utility/float_type.amd.hpp.in
+2
-0
driver/src/conv_driver.cpp
driver/src/conv_driver.cpp
+5
-3
No files found.
composable_kernel/include/utility/config.amd.hpp.in
View file @
3443835c
...
...
@@ -37,7 +37,7 @@
#endif
#ifndef CK_THREADWISE_GEMM_USE_AMD_INLINE_ASM
#define CK_THREADWISE_GEMM_USE_AMD_INLINE_ASM
0
#define CK_THREADWISE_GEMM_USE_AMD_INLINE_ASM
1
#endif
#ifndef CK_USE_AMD_V_FMAC_F32
...
...
composable_kernel/include/utility/float_type.amd.hpp.in
View file @
3443835c
...
...
@@ -322,6 +322,8 @@ struct inner_product_with_conversion
return acc;
}
__device__ T operator()(float_t a, float_t b) const { return convert(a) * convert(b); }
// hack for int8x4_t, because compiler does not have native support for int8x4_t
// int8x4_t is defined as int32_t
__device__ T operator()(int8x4_t a, int8x4_t b) const
...
...
driver/src/conv_driver.cpp
View file @
3443835c
...
...
@@ -49,7 +49,7 @@ int main(int argc, char* argv[])
using
LeftPads
=
Sequence
<
0
,
0
>
;
using
RightPads
=
Sequence
<
0
,
0
>
;
#elif
1
#elif
0
constexpr
index_t
N
=
1
;
constexpr
index_t
C
=
4
;
constexpr
index_t
HI
=
270
;
...
...
@@ -730,10 +730,12 @@ int main(int argc, char* argv[])
RightPads
{},
nrepeat
);
#elif 1
#if
0
#if
1
device_dynamic_convolution_forward_implicit_gemm_v4r4_nhwc_kyxc_nhwk
<
float
,
float
,
float
>
(
#elif 1
device_dynamic_convolution_forward_implicit_gemm_v4r4_nhwc_kyxc_nhwk
<
int8x4_t
,
int32_t
,
int32_t
>
(
device_dynamic_convolution_forward_implicit_gemm_v4r4_nhwc_kyxc_nhwk
<
int8x4_t
,
int32_t
,
int32_t
>
(
#elif 1
device_dynamic_convolution_forward_implicit_gemm_v4r4_nhwc_kyxc_nhwk
<
int8x4_t
,
int32_t
,
int8_t
>
(
#endif
...
...
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