Skip to content

Commit

Permalink
fix celu's alpha
Browse files Browse the repository at this point in the history
  • Loading branch information
hejunchao committed Aug 25, 2023
1 parent e7cdc86 commit 94b4296
Showing 1 changed file with 3 additions and 3 deletions.
6 changes: 3 additions & 3 deletions tests/kernels/test_celu.cpp
View file
Original file line number Diff line number Diff line change
Expand Up @@ -55,7 +55,7 @@ class CeluTest : public KernelTest,
case dt_float16: {
std::random_device rd;
std::mt19937 gen(rd());
std::uniform_real_distribution<float> dis(0.0f, 1.0f);
std::uniform_real_distribution<float> dis(0.0f, 5.0f);
NNCASE_UNUSED auto res = kernels::stackvm::apply(
tensor.shape(),
[&](gsl::span<const size_t> index) -> result<void> {
Expand All @@ -67,7 +67,7 @@ class CeluTest : public KernelTest,
case dt_float32: {
std::random_device rd;
std::mt19937 gen(rd());
std::uniform_real_distribution<float> dis(0.0f, 1.0f);
std::uniform_real_distribution<float> dis(0.0f, 5.0f);
NNCASE_UNUSED auto res = kernels::stackvm::apply(
tensor.shape(),
[&](gsl::span<const size_t> index) -> result<void> {
Expand All @@ -79,7 +79,7 @@ class CeluTest : public KernelTest,
case dt_bfloat16: {
std::random_device rd;
std::mt19937 gen(rd());
std::uniform_real_distribution<> dis(0.0, 1.0);
std::uniform_real_distribution<> dis(0.0, 5.0);
NNCASE_UNUSED auto res = kernels::stackvm::apply(
tensor.shape(),
[&](gsl::span<const size_t> index) -> result<void> {
Expand Down

0 comments on commit 94b4296

Please sign in to comment.