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_ROCM
Commits
3bec0b72
"...composable_kernel_rocm.git" did not exist on "a1b2441f8d3fc229629d0c6c18ef5836d1548e12"
Commit
3bec0b72
authored
Jan 03, 2025
by
mtgu0705
Browse files
fixed a bug in the ckProfiler, and formatted the code
parent
f52323fc
Changes
3
Hide whitespace changes
Inline
Side-by-side
Showing
3 changed files
with
5 additions
and
5 deletions
+5
-5
include/ck/utility/data_type.hpp
include/ck/utility/data_type.hpp
+1
-1
profiler/include/profiler/profile_gemm_b_scale_impl.hpp
profiler/include/profiler/profile_gemm_b_scale_impl.hpp
+1
-1
profiler/src/profile_gemm_b_scale.cpp
profiler/src/profile_gemm_b_scale.cpp
+3
-3
No files found.
include/ck/utility/data_type.hpp
View file @
3bec0b72
...
@@ -20,7 +20,7 @@ struct pk_i4_t
...
@@ -20,7 +20,7 @@ struct pk_i4_t
__host__
__device__
constexpr
pk_i4_t
()
:
data
{
type
{}}
{}
__host__
__device__
constexpr
pk_i4_t
()
:
data
{
type
{}}
{}
__host__
__device__
constexpr
pk_i4_t
(
type
init
)
:
data
{
init
}
{}
__host__
__device__
constexpr
pk_i4_t
(
type
init
)
:
data
{
init
}
{}
__host__
__device__
constexpr
operator
float
()
const
{
return
static_cast
<
floa
t
>
(
data
);
}
__host__
__device__
constexpr
operator
float
()
const
{
return
static_cast
<
int8_
t
>
(
data
);
}
};
};
inline
constexpr
auto
next_pow2
(
uint32_t
x
)
inline
constexpr
auto
next_pow2
(
uint32_t
x
)
...
...
profiler/include/profiler/profile_gemm_b_scale_impl.hpp
View file @
3bec0b72
...
@@ -342,7 +342,7 @@ bool profile_gemm_b_scale_impl(int do_verification,
...
@@ -342,7 +342,7 @@ bool profile_gemm_b_scale_impl(int do_verification,
if
(
do_log
)
if
(
do_log
)
{
{
LogRangeAsType
<
float
>
(
std
::
cout
<<
"a : "
,
a_m_k
.
mData
,
","
)
<<
std
::
endl
;
LogRangeAsType
<
float
>
(
std
::
cout
<<
"a : "
,
a_m_k
.
mData
,
","
)
<<
std
::
endl
;
LogRangeAsType
<
floa
t
>
(
std
::
cout
<<
"b: "
,
b_k_n
.
mData
,
","
)
<<
std
::
endl
;
LogRangeAsType
<
int8_
t
>
(
std
::
cout
<<
"b: "
,
b_k_n
.
mData
,
","
)
<<
std
::
endl
;
LogRangeAsType
<
float
>
(
LogRangeAsType
<
float
>
(
std
::
cout
<<
"c_host : "
,
c_m_n_host_result
.
mData
,
","
)
std
::
cout
<<
"c_host : "
,
c_m_n_host_result
.
mData
,
","
)
<<
std
::
endl
;
<<
std
::
endl
;
...
...
profiler/src/profile_gemm_b_scale.cpp
View file @
3bec0b72
...
@@ -103,9 +103,9 @@ int profile_gemm_b_scale(int argc, char* argv[])
...
@@ -103,9 +103,9 @@ int profile_gemm_b_scale(int argc, char* argv[])
printf
(
"n_warmup:%d, n_iter:%d, rotating:%lu
\n
"
,
n_warmup
,
n_iter
,
rotating
);
printf
(
"n_warmup:%d, n_iter:%d, rotating:%lu
\n
"
,
n_warmup
,
n_iter
,
rotating
);
}
}
using
F32
=
float
;
using
F32
=
float
;
using
F16
=
ck
::
half_t
;
using
F16
=
ck
::
half_t
;
using
I4
=
ck
::
pk_i4_t
;
using
I4
=
ck
::
pk_i4_t
;
using
Row
=
ck
::
tensor_layout
::
gemm
::
RowMajor
;
using
Row
=
ck
::
tensor_layout
::
gemm
::
RowMajor
;
using
Col
=
ck
::
tensor_layout
::
gemm
::
ColumnMajor
;
using
Col
=
ck
::
tensor_layout
::
gemm
::
ColumnMajor
;
...
...
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