From fc0084f4a6008b3b655119dac0f6c31dd486f0d9 Mon Sep 17 00:00:00 2001
From: liuqi <liuqi10@xiaomi.com>
Date: Wed, 7 Mar 2018 11:00:32 +0800
Subject: [PATCH] Replace MACE_CHECK with MACE_CHECK_NOTNULL.

---
 mace/kernels/activation.h                | 2 +-
 mace/kernels/opencl/activation_opencl.cc | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/mace/kernels/activation.h b/mace/kernels/activation.h
index c9f8dac5..72e52b67 100644
--- a/mace/kernels/activation.h
+++ b/mace/kernels/activation.h
@@ -116,7 +116,7 @@ class ActivationFunctor {
     const T *input_ptr = input->data<T>();
     T *output_ptr = output->mutable_data<T>();
     if (activation_ == PRELU) {
-      MACE_CHECK(alpha != nullptr) << "PReLU's alpha parameter shouldn't be null";
+      MACE_CHECK_NOTNULL(alpha);
       const T *alpha_ptr = alpha->data<T>();
       PReLUActivation(input_ptr, output->size(), input->dim(3), alpha_ptr, output_ptr); 
     } else {
diff --git a/mace/kernels/opencl/activation_opencl.cc b/mace/kernels/opencl/activation_opencl.cc
index 9521f82e..75922a9e 100644
--- a/mace/kernels/opencl/activation_opencl.cc
+++ b/mace/kernels/opencl/activation_opencl.cc
@@ -62,7 +62,7 @@ void ActivationFunctor<DeviceType::OPENCL, T>::operator()(const Tensor *input,
     int idx = 0;
     kernel_.setArg(idx++, *(input->opencl_image()));
     if (activation_ == PRELU) {
-      MACE_CHECK(alpha != nullptr) << "PReLU's alpha parameter shouldn't be null";
+      MACE_CHECK_NOTNULL(alpha);
       kernel_.setArg(idx++, *(alpha->opencl_image()));
     }
     kernel_.setArg(idx++, static_cast<float>(relux_max_limit_));
-- 
GitLab