huh

llama/src/llama.jn

+fn embedding_lookup<T:number, num_embeddings:usize, embedding_dim:usize, query_dim:usize>(
+  weights: T[num_embeddings, embedding_dim],
+  query: usize[query_dim]
+) -> T[query_dim, embedding_dim] {
+  let res : T[query_dim, embedding_dim];
+  for q = 0 to query_dim {
+    for e = 0 to embedding_dim {
+      res[q,e] = weights[q,e];
+    }
+  }
+  return res;
+}
+
 #[entry]
-fn llama() -> i32 {
-  return 1;
+fn llama_f32<num_embeddings: usize, embedding_dim: usize>(weights_embedding: f32[num_embeddings, embedding_dim]) -> f32 {
+  let query: usize[1];
+  query[0] = 0;
+  return embedding_lookup::<f32, num_embeddings, embedding_dim, 1>(weights_embedding, query)[0,0];
 }

llama/src/main.rs

@@ -2,11 +2,11 @@
 #![feature(entry_insert)]
 
 use hercules_rt::runner;
-juno_build::juno!("llama");
+juno_build::juno!("llama_f32");
 
 fn main() {
     async_std::task::block_on(async {
-        let mut r = runner!(llama);
+        let mut r = runner!(llama_f32);
         let result = r.run().await;
         assert_eq!(result, 1);
     });