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
c3c89727
Commit
c3c89727
authored
Jun 25, 2023
by
carlushuang
Browse files
fix CI warning
parent
50e573f5
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
10 additions
and
6 deletions
+10
-6
example/01_gemm/run_gemm_example.inc
example/01_gemm/run_gemm_example.inc
+10
-6
No files found.
example/01_gemm/run_gemm_example.inc
View file @
c3c89727
...
...
@@ -61,16 +61,20 @@ bool run_gemm(const ProblemType& problem_size, const ExecutionConfig& config)
switch
(
config
.
init_method
)
{
case
0
:
ck
::
utils
::
FillConstant
<
ADataType
>
{
1.
f
}(
a_m_k
);
ck
::
utils
::
FillConstant
<
BDataType
>
{
1.
f
}(
b_k_n
);
ck
::
utils
::
FillConstant
<
ADataType
>
{
static_cast
<
ADataType
>
(
1.
f
)
}(
a_m_k
);
ck
::
utils
::
FillConstant
<
BDataType
>
{
static_cast
<
BDataType
>
(
1.
f
)
}(
b_k_n
);
break
;
case
1
:
ck
::
utils
::
FillUniformDistributionIntegerValue
<
ADataType
>
{
-
5.
f
,
5.
f
}(
a_m_k
);
ck
::
utils
::
FillUniformDistributionIntegerValue
<
BDataType
>
{
-
5.
f
,
5.
f
}(
b_k_n
);
ck
::
utils
::
FillUniformDistributionIntegerValue
<
ADataType
>
{
static_cast
<
ADataType
>
(
-
5.
f
),
static_cast
<
ADataType
>
(
5.
f
)}(
a_m_k
);
ck
::
utils
::
FillUniformDistributionIntegerValue
<
BDataType
>
{
static_cast
<
BDataType
>
(
-
5.
f
),
static_cast
<
BDataType
>
(
5.
f
)}(
b_k_n
);
break
;
default
:
ck
::
utils
::
FillUniformDistribution
<
ADataType
>
{
-
1.
f
,
1.
f
}(
a_m_k
);
ck
::
utils
::
FillUniformDistribution
<
BDataType
>
{
-
1.
f
,
1.
f
}(
b_k_n
);
ck
::
utils
::
FillUniformDistribution
<
ADataType
>
{
static_cast
<
ADataType
>
(
-
1.
f
),
static_cast
<
ADataType
>
(
1.
f
)}(
a_m_k
);
ck
::
utils
::
FillUniformDistribution
<
BDataType
>
{
static_cast
<
BDataType
>
(
-
1.
f
),
static_cast
<
BDataType
>
(
1.
f
)}(
b_k_n
);
}
Tensor
<
CDataType
>
c_m_n_host_result
(
f_host_tensor_descriptor
(
M
,
N
,
StrideC
,
CLayout
{}));
...
...
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