ember.cpp 0.1.0
C++ inference engine for Llama-family and paperkiln models (CPU, CUDA, WebAssembly)
Loading...
Searching...
No Matches
graph_ir_expr.h File Reference
#include <cstddef>
#include <string>
#include <vector>
Include dependency graph for graph_ir_expr.h:
This graph shows which files directly or indirectly include this file:

Go to the source code of this file.

Classes

struct  ember_ir::expr::Node
 
class  ember_ir::expr::Expr
 
struct  ember_ir::expr::Instr
 
struct  ember_ir::expr::Stage
 

Namespaces

namespace  ember_ir
 
namespace  ember_ir::expr
 

Enumerations

enum class  ember_ir::expr::Kind {
  ember_ir::expr::NUMBER , ember_ir::expr::INPUT , ember_ir::expr::NEG , ember_ir::expr::ADD ,
  ember_ir::expr::SUB , ember_ir::expr::MUL , ember_ir::expr::DIV , ember_ir::expr::POW ,
  ember_ir::expr::EXP , ember_ir::expr::LOG , ember_ir::expr::SQRT , ember_ir::expr::TANH ,
  ember_ir::expr::SIGMOID , ember_ir::expr::ERF , ember_ir::expr::ABS , ember_ir::expr::RELU ,
  ember_ir::expr::MAX , ember_ir::expr::MIN , ember_ir::expr::WHERE , ember_ir::expr::ROWSUM ,
  ember_ir::expr::ROWMEAN , ember_ir::expr::ROWMAX
}
 
enum class  ember_ir::expr::Op : int {
  ember_ir::expr::CONST , ember_ir::expr::INPUT , ember_ir::expr::COL , ember_ir::expr::WIDTH ,
  ember_ir::expr::NEG , ember_ir::expr::EXP , ember_ir::expr::LOG , ember_ir::expr::SQRT ,
  ember_ir::expr::TANH , ember_ir::expr::SIGMOID , ember_ir::expr::ERF , ember_ir::expr::ABS ,
  ember_ir::expr::RELU , ember_ir::expr::ADD , ember_ir::expr::SUB , ember_ir::expr::MUL ,
  ember_ir::expr::DIV , ember_ir::expr::POW , ember_ir::expr::MAX , ember_ir::expr::MIN ,
  ember_ir::expr::WHERE
}
 

Functions

void ember_ir::expr::eval (const Expr &e, const std::vector< const float * > &ins, int T, int w, float *out)
 
const Expr & ember_ir::expr::cached (const std::string &text, size_t n_inputs)
 
std::vector< Stage > ember_ir::expr::lower (const Expr &e)
 

Variables

constexpr size_t ember_ir::expr::kMaxCode = 96
 
constexpr size_t ember_ir::expr::kMaxStack = 16
 
constexpr size_t ember_ir::expr::kMaxColumns = 8