n_sign.cc

Go to the documentation of this file.

/* FEAT
copyright 2017 William La Cava
license: GNU/GPL v3
*/
 
#include "n_sign.h"
 
namespace FT{
 
    namespace Pop{
        namespace Op{
            NodeSign::NodeSign()
            {
                name = "sign";
                otype = 'f';
                arity['f'] = 1;
                complexity = 1;
 
            }
 
            #ifndef USE_CUDA
            void NodeSign::evaluate(const Data& data, State& state)
            {
                ArrayXf x = state.pop<float>();
                ArrayXf ones = ArrayXf::Ones(x.size());
 
                ArrayXf res = ( x > 0).select(ones, 
                                                    (x == 0).select(ArrayXf::Zero(x.size()), 
                                                                    -1*ones)); 
                state.push<float>(res);
            }
            #else
            void NodeSign::evaluate(const Data& data, State& state)
            {
                GPU_Sign(state.dev_f, state.idx[otype], state.N);
            }
            #endif
 
            void NodeSign::eval_eqn(State& state)
            {
                state.push<float>("sign("+ state.popStr<float>() +")");
            }
 
            
            NodeSign* NodeSign::clone_impl() const { return new NodeSign(*this); }
 
            NodeSign* NodeSign::rnd_clone_impl() const { return new NodeSign(); }  
        }
    }
}