Rust-related/luminal
1
0
Fork 1
mirror of https://github.com/jafioti/luminal.git synced 2026-06-01 21:49:47 +09:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #123 from anuragsingh-tt/anuhsing/minor_fix

Browse Source
This commit is contained in:
Joe Fioti
2025-07-16 21:15:08 -05:00
committed by GitHub
parent 73299b535e 9c2754d42d
commit d6901cc018
2 changed files with 3 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
examples/llama/README.md
View File

@@ -5,7 +5,7 @@
## step-by-step instructions:
1. Download Llama-3 8B weights from Hugging Face
- ```bash
bash setup.sh
bash setup/setup.sh
```
2. Execute a Sample Llama Prompt
- if you are on a macbook:

3
flash_attention_demo/src/main.rs
View File

@@ -31,7 +31,8 @@ pub fn run_egglog_program(code: &str) -> Result<(Vec<String>, String, String), E
let msgs = egraph.run_program(commands)?;
println!("Run Report: {}", egraph.get_run_report().as_ref().unwrap());
let (sort, value) = egraph.eval_expr(&var!("one_output"))?;
let (termdag, root) = egraph.extract_value(&sort, value)?;
let (termdag, root, _) = egraph.extract_value(&sort, value)?;
// Remove the underscore prefix and uncomment the display_graph call to display the graph in GraphViz.
let (_petgraph, _root_idx) = dag_to_petgraph(&termdag, root.clone());
// display_graph(&petgraph, &[root_idx]);
let s = egraph.serialize(egglog::SerializeConfig {