diff --git a/candle-kernels/src/lib.rs b/candle-kernels/src/lib.rs index 8b13789179..478debd3d4 100644 --- a/candle-kernels/src/lib.rs +++ b/candle-kernels/src/lib.rs @@ -1 +1,9 @@ - +pub const AFFINE: &str = include_str!(concat!(env!("OUT_DIR"), "/affine.ptx")); +pub const BINARY: &str = include_str!(concat!(env!("OUT_DIR"), "/binary.ptx")); +pub const CAST: &str = include_str!(concat!(env!("OUT_DIR"), "/cast.ptx")); +pub const CONV: &str = include_str!(concat!(env!("OUT_DIR"), "/conv.ptx")); +pub const FILL: &str = include_str!(concat!(env!("OUT_DIR"), "/fill.ptx")); +pub const INDEXING: &str = include_str!(concat!(env!("OUT_DIR"), "/indexing.ptx")); +pub const REDUCE: &str = include_str!(concat!(env!("OUT_DIR"), "/reduce.ptx")); +pub const TERNARY: &str = include_str!(concat!(env!("OUT_DIR"), "/ternary.ptx")); +pub const UNARY: &str = include_str!(concat!(env!("OUT_DIR"), "/unary.ptx"));