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
ddd42c2d
Commit
ddd42c2d
authored
Feb 17, 2025
by
Jakub Piasecki
Browse files
added fp8 and bf8 instances
parent
478dfe12
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
6 additions
and
6 deletions
+6
-6
example/ck_tile/03_gemm/gemm.hpp
example/ck_tile/03_gemm/gemm.hpp
+6
-6
No files found.
example/ck_tile/03_gemm/gemm.hpp
View file @
ddd42c2d
...
@@ -89,7 +89,7 @@ struct DataTypeTraits<ck_tile::bf8_t>
...
@@ -89,7 +89,7 @@ struct DataTypeTraits<ck_tile::bf8_t>
static
constexpr
const
char
*
name
=
"bf8"
;
static
constexpr
const
char
*
name
=
"bf8"
;
};
};
/**
\
brief Struct used for specifying desired gemm details*/
/**
@
brief Struct used for specifying desired gemm details*/
struct
gemm_traits
struct
gemm_traits
{
{
std
::
string
data_type
;
/** Tensors datatype, can be set to either fp16 or bf16*/
std
::
string
data_type
;
/** Tensors datatype, can be set to either fp16 or bf16*/
...
@@ -146,12 +146,12 @@ template <typename Traits_>
...
@@ -146,12 +146,12 @@ template <typename Traits_>
float
gemm_
(
const
ck_tile
::
GemmHostArgs
&
args
,
const
ck_tile
::
stream_config
&
s
);
float
gemm_
(
const
ck_tile
::
GemmHostArgs
&
args
,
const
ck_tile
::
stream_config
&
s
);
/**
/**
*
\
brief Invoke gemm function
*
@
brief Invoke gemm function
*
*
*
\
param traits Gemm traits which are used for choosing best instance.
*
@
param traits Gemm traits which are used for choosing best instance.
*
\
param args Runtime gemm host arguments.
*
@
param args Runtime gemm host arguments.
*
\
param s Stream configuration.
*
@
param s Stream configuration.
*
\
return Time of execution.
*
@
return Time of execution.
*/
*/
float
gemm
(
const
gemm_traits
&
traits
,
float
gemm
(
const
gemm_traits
&
traits
,
const
ck_tile
::
GemmHostArgs
&
args
,
const
ck_tile
::
GemmHostArgs
&
args
,
...
...
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