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

Go to the source code of this file.

Namespaces

namespace  ember_ir
 
namespace  ember_ir::cpu
 

Functions

void ember_ir::cpu::conv1d_causal_row (const float *x, const float *w, const float *bias, float *state, float *out, int C, int K)
 
void ember_ir::cpu::selective_scan_row (const float *x, const float *dt, const float *B, const float *C, const float *A, const float *d, float *s, float *y, int D, int N, bool softplus)