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
68e3bb6d
"tests/vscode:/vscode.git/clone" did not exist on "cd21b965d1d10266d399b887f77b7dfc429f5f04"
Commit
68e3bb6d
authored
Jul 15, 2023
by
danyao12
Browse files
recover data_type.hpp
parent
6d6520ab
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
0 additions
and
14 deletions
+0
-14
include/ck/utility/data_type.hpp
include/ck/utility/data_type.hpp
+0
-14
No files found.
include/ck/utility/data_type.hpp
View file @
68e3bb6d
...
...
@@ -965,19 +965,6 @@ inline __host__ __device__ constexpr float type_convert<float, bhalf_t>(bhalf_t
}
// convert fp32 to bfp16
#if FLASH_ATTENTION_INTERNAL_USE_RTZ
template
<
>
inline
__host__
__device__
constexpr
bhalf_t
type_convert
<
bhalf_t
,
float
>
(
float
x
)
{
union
{
float
fp32
;
uint32_t
int32
;
}
u
=
{
static_cast
<
float
>
(
x
)};
return
uint16_t
(
u
.
int32
>>
16
);
}
#else
template
<
>
inline
__host__
__device__
constexpr
bhalf_t
type_convert
<
bhalf_t
,
float
>
(
float
x
)
{
...
...
@@ -1020,7 +1007,6 @@ inline __host__ __device__ constexpr bhalf_t type_convert<bhalf_t, float>(float
return
uint16_t
(
u
.
int32
>>
16
);
}
#endif
// convert fp16 to bf16
template
<
>
...
...
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