ROCm/composable_kernel
1
0
Fork 0
mirror of https://github.com/ROCm/composable_kernel.git synced 2026-05-12 09:16:52 +00:00
Code Issues Packages Projects Releases Wiki Activity

Use double for all scaling values and float-point constant values at the Device Op API (#557)

Browse Source 
* Use double as alpha/beta values type in reduce device op api

* Use double as alpha/beta values type in softmax device op api

* Use double as alpha/beta values type in multiple-reduce device op api

* Use double as epsilon value type in normalization/elementwise-normalization device op api
This commit is contained in:
Qianfeng
2023-01-19 02:02:50 +08:00
committed by GitHub
parent 1cfa87608a
commit 52abc2f371
24 changed files with 112 additions and 109 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
profiler/src/profile_softmax.cpp
View File

@@ -99,8 +99,8 @@ int profile_softmax(int argc, char* argv[])
length,
stride,
reduce,
float(alpha),
float(beta));
double(alpha),
double(beta));
}
else if(data_type == SoftmaxDataType::F32_F32)
{
@@ -111,8 +111,8 @@ int profile_softmax(int argc, char* argv[])
length,
stride,
reduce,
float(alpha),
float(beta));
double(alpha),
double(beta));
}
else
{
@@ -131,8 +131,8 @@ int profile_softmax(int argc, char* argv[])
length,
stride,
reduce,
float(alpha),
float(beta));
double(alpha),
double(beta));
}
else if(data_type == SoftmaxDataType::F32_F32)
{
@@ -143,8 +143,8 @@ int profile_softmax(int argc, char* argv[])
length,
stride,
reduce,
float(alpha),
float(beta));
double(alpha),
double(beta));
}
else
{