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
yangql
composable_kernel-1
Commits
cb954213
"git@developer.sourcefind.cn:modelzoo/resnet50_tensorflow.git" did not exist on "6941537ba4882e2ed4a33da0cd5706e3e35eadbb"
Commit
cb954213
authored
Aug 06, 2021
by
Chao Liu
Browse files
refactor
parent
73ca9701
Changes
7
Hide whitespace changes
Inline
Side-by-side
Showing
7 changed files
with
20 additions
and
13 deletions
+20
-13
host/driver_online/conv_fwd_driver_online.cpp
host/driver_online/conv_fwd_driver_online.cpp
+1
-1
host/driver_online/include/online_device_dynamic_convolution_forward_implicit_gemm_v4r4_dlops_nchw_kcyx_nkhw.hpp
...ution_forward_implicit_gemm_v4r4_dlops_nchw_kcyx_nkhw.hpp
+1
-1
host/driver_online/include/online_device_dynamic_convolution_forward_implicit_gemm_v4r4_xdlops_nchw_kcyx_nkhw.hpp
...tion_forward_implicit_gemm_v4r4_xdlops_nchw_kcyx_nkhw.hpp
+1
-1
host/driver_online/include/online_device_dynamic_convolution_forward_implicit_gemm_v6r1_dlops_nchw_kcyx_nkhw.hpp
...ution_forward_implicit_gemm_v6r1_dlops_nchw_kcyx_nkhw.hpp
+5
-4
host/driver_online/include/online_driver_common.hpp
host/driver_online/include/online_driver_common.hpp
+4
-2
host/solver/include/conv_igemm_fwd_v6r1_dlops_nchw_kcyx_nkhw.hpp
...lver/include/conv_igemm_fwd_v6r1_dlops_nchw_kcyx_nkhw.hpp
+4
-2
host/solver/include/convolution_problem_descriptor.hpp
host/solver/include/convolution_problem_descriptor.hpp
+4
-2
No files found.
host/driver_online/conv_fwd_driver_online.cpp
View file @
cb954213
...
@@ -35,7 +35,7 @@ enum ConvForwardAlgo
...
@@ -35,7 +35,7 @@ enum ConvForwardAlgo
int
main
(
int
argc
,
char
*
argv
[])
int
main
(
int
argc
,
char
*
argv
[])
{
{
using
namespace
ck
;
using
namespace
ck
;
using
namespace
ck
_
driver
;
using
namespace
ck
::
driver
;
using
size_t
=
std
::
size_t
;
using
size_t
=
std
::
size_t
;
hipStream_t
stream
;
hipStream_t
stream
;
...
...
host/driver_online/include/online_device_dynamic_convolution_forward_implicit_gemm_v4r4_dlops_nchw_kcyx_nkhw.hpp
View file @
cb954213
...
@@ -231,7 +231,7 @@ void online_device_dynamic_convolution_forward_implicit_gemm_v4r4_dlops_nchw_kcy
...
@@ -231,7 +231,7 @@ void online_device_dynamic_convolution_forward_implicit_gemm_v4r4_dlops_nchw_kcy
ck
::
index_t
nrepeat
)
ck
::
index_t
nrepeat
)
{
{
using
namespace
ck
;
using
namespace
ck
;
using
namespace
ck
_
driver
;
using
namespace
ck
::
driver
;
using
namespace
detail_dyn_conv_fwd_v4r4_nchw_kcyx_nkhw
;
using
namespace
detail_dyn_conv_fwd_v4r4_nchw_kcyx_nkhw
;
using
size_t
=
std
::
size_t
;
using
size_t
=
std
::
size_t
;
...
...
host/driver_online/include/online_device_dynamic_convolution_forward_implicit_gemm_v4r4_xdlops_nchw_kcyx_nkhw.hpp
View file @
cb954213
...
@@ -227,7 +227,7 @@ void online_device_dynamic_convolution_forward_implicit_gemm_v4r4_xdlops_nchw_kc
...
@@ -227,7 +227,7 @@ void online_device_dynamic_convolution_forward_implicit_gemm_v4r4_xdlops_nchw_kc
ck
::
index_t
nrepeat
)
ck
::
index_t
nrepeat
)
{
{
using
namespace
ck
;
using
namespace
ck
;
using
namespace
ck
_
driver
;
using
namespace
ck
::
driver
;
using
namespace
detail_dyn_conv_fwd_v4r4_xdlops_nchw_kcyx_nkhw
;
using
namespace
detail_dyn_conv_fwd_v4r4_xdlops_nchw_kcyx_nkhw
;
using
size_t
=
std
::
size_t
;
using
size_t
=
std
::
size_t
;
...
...
host/driver_online/include/online_device_dynamic_convolution_forward_implicit_gemm_v6r1_dlops_nchw_kcyx_nkhw.hpp
View file @
cb954213
...
@@ -31,11 +31,11 @@ void online_device_dynamic_convolution_forward_implicit_gemm_v6r1_dlops_nchw_kcy
...
@@ -31,11 +31,11 @@ void online_device_dynamic_convolution_forward_implicit_gemm_v6r1_dlops_nchw_kcy
const
Tensor
<
TInWei
>&
in_n_c_hi_wi
,
const
Tensor
<
TInWei
>&
in_n_c_hi_wi
,
const
Tensor
<
TInWei
>&
wei_k_c_y_x
,
const
Tensor
<
TInWei
>&
wei_k_c_y_x
,
Tensor
<
TOut
>&
out_n_k_ho_wo
,
Tensor
<
TOut
>&
out_n_k_ho_wo
,
const
ck
_
driver
::
CompileParameterConvIgemmFwdV6r1DlopsNchwKcyxNkhw
&
compile_param
,
const
ck
::
driver
::
CompileParameterConvIgemmFwdV6r1DlopsNchwKcyxNkhw
&
compile_param
,
ck
::
index_t
nrepeat
)
ck
::
index_t
nrepeat
)
{
{
using
namespace
ck
;
using
namespace
ck
;
using
namespace
ck
_
driver
;
using
namespace
ck
::
driver
;
using
size_t
=
std
::
size_t
;
using
size_t
=
std
::
size_t
;
std
::
cout
<<
__func__
<<
std
::
endl
;
std
::
cout
<<
__func__
<<
std
::
endl
;
...
@@ -100,8 +100,9 @@ void online_device_dynamic_convolution_forward_implicit_gemm_v6r1_dlops_nchw_kcy
...
@@ -100,8 +100,9 @@ void online_device_dynamic_convolution_forward_implicit_gemm_v6r1_dlops_nchw_kcy
"dynamic_convolution_forward_implicit_gemm_v6r1_dlops_nchw_kcyx_nkhw.cpp"
;
"dynamic_convolution_forward_implicit_gemm_v6r1_dlops_nchw_kcyx_nkhw.cpp"
;
std
::
string
algo_name
=
"implicit_gemm_conv_fwd_v6r1_dlops_nchw"
;
std
::
string
algo_name
=
"implicit_gemm_conv_fwd_v6r1_dlops_nchw"
;
std
::
string
compile_param_string
=
get_ck_hip_online_compile_common_flag
()
+
compile_param
.
GetCompileParameterString
();
std
::
string
compile_param_string
=
std
::
string
network_config
=
compile_param_string
;
get_ck_hip_online_compile_common_flag
()
+
compile_param
.
GetCompileParameterString
();
std
::
string
network_config
=
compile_param_string
;
std
::
vector
<
float
>
kernel1_times
;
std
::
vector
<
float
>
kernel1_times
;
std
::
vector
<
float
>
kernel2_times
;
std
::
vector
<
float
>
kernel2_times
;
...
...
host/driver_online/include/online_driver_common.hpp
View file @
cb954213
#ifndef ONLINE_DRIVER_COMMON_HPP
#ifndef ONLINE_DRIVER_COMMON_HPP
#define ONLINE_DRIVER_COMMON_HPP
#define ONLINE_DRIVER_COMMON_HPP
namespace
ck_driver
{
namespace
ck
{
namespace
driver
{
inline
auto
get_ck_hip_online_compile_common_flag
()
inline
auto
get_ck_hip_online_compile_common_flag
()
{
{
...
@@ -47,5 +48,6 @@ auto gcd(X x, Ys... ys)
...
@@ -47,5 +48,6 @@ auto gcd(X x, Ys... ys)
return
gcd
(
x
,
gcd
(
ys
...));
return
gcd
(
x
,
gcd
(
ys
...));
}
}
}
// namespace ck_driver
}
// namespace driver
}
// namespace ck
#endif
#endif
host/solver/include/conv_igemm_fwd_v6r1_dlops_nchw_kcyx_nkhw.hpp
View file @
cb954213
...
@@ -3,7 +3,8 @@
...
@@ -3,7 +3,8 @@
#include <numeric>
#include <numeric>
namespace
ck_driver
{
namespace
ck
{
namespace
driver
{
struct
CompileParameterConvIgemmFwdV6r1DlopsNchwKcyxNkhw
struct
CompileParameterConvIgemmFwdV6r1DlopsNchwKcyxNkhw
{
{
...
@@ -669,5 +670,6 @@ struct ConvIgemmFwdV6r1DlopsNchwKcyxNkhw
...
@@ -669,5 +670,6 @@ struct ConvIgemmFwdV6r1DlopsNchwKcyxNkhw
}
}
};
};
}
// namespace ck_driver
}
// namespace driver
}
// namespace ck
#endif
#endif
host/solver/include/convolution_problem_descriptor.hpp
View file @
cb954213
#ifndef CONVOLUTION_PROBLEM_DESCRIPTOR
#ifndef CONVOLUTION_PROBLEM_DESCRIPTOR
#define CONVOLUTION_PROBLEM_DESCRIPTOR
#define CONVOLUTION_PROBLEM_DESCRIPTOR
namespace
ck_driver
{
namespace
ck
{
namespace
driver
{
struct
ConvolutionProblemDescriptor
struct
ConvolutionProblemDescriptor
{
{
...
@@ -75,5 +76,6 @@ struct ConvolutionProblemDescriptor
...
@@ -75,5 +76,6 @@ struct ConvolutionProblemDescriptor
std
::
size_t
CalculateFlop
()
const
{
return
2L
*
N
*
K
*
C
*
Y
*
X
*
Ho
*
Wo
;
}
std
::
size_t
CalculateFlop
()
const
{
return
2L
*
N
*
K
*
C
*
Y
*
X
*
Ho
*
Wo
;
}
};
};
}
// namespace ck_driver
}
// namespace driver
}
// namespace ck
#endif
#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