├── .clang-format ├── .clang-tidy ├── .env ├── .gitignore ├── CMakeLists.txt ├── LICENSE.txt ├── README.md ├── eq_sat.sh ├── src ├── EggifyPass.h ├── Egglog.cpp ├── Egglog.h ├── EgglogCustomDefs.h ├── EqualitySaturationPass.cpp ├── EqualitySaturationPass.h ├── Utils.h ├── base.egg └── egg-opt.cpp └── test ├── 2mm ├── 2mm.egg └── 2mm.mlir ├── 3mm ├── 3mm.egg └── 3mm.mlir ├── CMakeLists.txt ├── arith └── arith_tensor.mlir ├── classic ├── classic.egg └── classic.mlir ├── clean.sh ├── constant_folding ├── constant_folding.egg └── constant_folding.mlir ├── image_conversion ├── image_conversion.egg └── image_conversion.mlir ├── interpret.sh ├── lit.cfg.py ├── lit.site.cfg.py.in ├── nmm ├── 10mm.mlir ├── 160mm.mlir ├── 160sqmm.mlir ├── 20mm.mlir ├── 2mm.mlir ├── 3mm.mlir ├── 40mm.mlir ├── 40sqmm.mlir ├── 4mm.mlir ├── 5mm.mlir ├── 80mm.mlir ├── 80sqmm.mlir └── nmm.egg ├── polynomial ├── polynomial.egg └── polynomial.mlir ├── run.sh ├── scf ├── scf_if.egg └── scf_if.mlir ├── stablehlo ├── tests.egg └── tests.mlir ├── stablehlo_benchmarks ├── gmm.csv ├── gmm.mlir ├── jaxpt.csv ├── kex.csv ├── kex.mlir ├── llama.csv ├── nasrnn.csv ├── nasrnn.mlir ├── resnet.csv ├── rewrites.txt ├── stablehlo.csv ├── stablehlo.egg └── stablehlo.min.egg ├── util ├── gen_libutil.sh └── util.mlir └── vector_norm ├── vector_norm.egg └── vector_norm.mlir /.clang-format: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AzizZayed/DialEgg/HEAD/.clang-format -------------------------------------------------------------------------------- /.clang-tidy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AzizZayed/DialEgg/HEAD/.clang-tidy -------------------------------------------------------------------------------- /.env: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AzizZayed/DialEgg/HEAD/.env -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AzizZayed/DialEgg/HEAD/.gitignore -------------------------------------------------------------------------------- /CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AzizZayed/DialEgg/HEAD/CMakeLists.txt -------------------------------------------------------------------------------- /LICENSE.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AzizZayed/DialEgg/HEAD/LICENSE.txt -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AzizZayed/DialEgg/HEAD/README.md -------------------------------------------------------------------------------- /eq_sat.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AzizZayed/DialEgg/HEAD/eq_sat.sh -------------------------------------------------------------------------------- /src/EggifyPass.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AzizZayed/DialEgg/HEAD/src/EggifyPass.h -------------------------------------------------------------------------------- /src/Egglog.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AzizZayed/DialEgg/HEAD/src/Egglog.cpp -------------------------------------------------------------------------------- /src/Egglog.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AzizZayed/DialEgg/HEAD/src/Egglog.h -------------------------------------------------------------------------------- /src/EgglogCustomDefs.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AzizZayed/DialEgg/HEAD/src/EgglogCustomDefs.h -------------------------------------------------------------------------------- /src/EqualitySaturationPass.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AzizZayed/DialEgg/HEAD/src/EqualitySaturationPass.cpp -------------------------------------------------------------------------------- /src/EqualitySaturationPass.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AzizZayed/DialEgg/HEAD/src/EqualitySaturationPass.h -------------------------------------------------------------------------------- /src/Utils.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AzizZayed/DialEgg/HEAD/src/Utils.h -------------------------------------------------------------------------------- /src/base.egg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AzizZayed/DialEgg/HEAD/src/base.egg -------------------------------------------------------------------------------- /src/egg-opt.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AzizZayed/DialEgg/HEAD/src/egg-opt.cpp -------------------------------------------------------------------------------- /test/2mm/2mm.egg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AzizZayed/DialEgg/HEAD/test/2mm/2mm.egg -------------------------------------------------------------------------------- /test/2mm/2mm.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AzizZayed/DialEgg/HEAD/test/2mm/2mm.mlir -------------------------------------------------------------------------------- /test/3mm/3mm.egg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AzizZayed/DialEgg/HEAD/test/3mm/3mm.egg -------------------------------------------------------------------------------- /test/3mm/3mm.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AzizZayed/DialEgg/HEAD/test/3mm/3mm.mlir -------------------------------------------------------------------------------- /test/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AzizZayed/DialEgg/HEAD/test/CMakeLists.txt -------------------------------------------------------------------------------- /test/arith/arith_tensor.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AzizZayed/DialEgg/HEAD/test/arith/arith_tensor.mlir -------------------------------------------------------------------------------- /test/classic/classic.egg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AzizZayed/DialEgg/HEAD/test/classic/classic.egg -------------------------------------------------------------------------------- /test/classic/classic.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AzizZayed/DialEgg/HEAD/test/classic/classic.mlir -------------------------------------------------------------------------------- /test/clean.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AzizZayed/DialEgg/HEAD/test/clean.sh -------------------------------------------------------------------------------- /test/constant_folding/constant_folding.egg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AzizZayed/DialEgg/HEAD/test/constant_folding/constant_folding.egg -------------------------------------------------------------------------------- /test/constant_folding/constant_folding.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AzizZayed/DialEgg/HEAD/test/constant_folding/constant_folding.mlir -------------------------------------------------------------------------------- /test/image_conversion/image_conversion.egg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AzizZayed/DialEgg/HEAD/test/image_conversion/image_conversion.egg -------------------------------------------------------------------------------- /test/image_conversion/image_conversion.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AzizZayed/DialEgg/HEAD/test/image_conversion/image_conversion.mlir -------------------------------------------------------------------------------- /test/interpret.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AzizZayed/DialEgg/HEAD/test/interpret.sh -------------------------------------------------------------------------------- /test/lit.cfg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AzizZayed/DialEgg/HEAD/test/lit.cfg.py -------------------------------------------------------------------------------- /test/lit.site.cfg.py.in: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AzizZayed/DialEgg/HEAD/test/lit.site.cfg.py.in -------------------------------------------------------------------------------- /test/nmm/10mm.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AzizZayed/DialEgg/HEAD/test/nmm/10mm.mlir -------------------------------------------------------------------------------- /test/nmm/160mm.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AzizZayed/DialEgg/HEAD/test/nmm/160mm.mlir -------------------------------------------------------------------------------- /test/nmm/160sqmm.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AzizZayed/DialEgg/HEAD/test/nmm/160sqmm.mlir -------------------------------------------------------------------------------- /test/nmm/20mm.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AzizZayed/DialEgg/HEAD/test/nmm/20mm.mlir -------------------------------------------------------------------------------- /test/nmm/2mm.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AzizZayed/DialEgg/HEAD/test/nmm/2mm.mlir -------------------------------------------------------------------------------- /test/nmm/3mm.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AzizZayed/DialEgg/HEAD/test/nmm/3mm.mlir -------------------------------------------------------------------------------- /test/nmm/40mm.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AzizZayed/DialEgg/HEAD/test/nmm/40mm.mlir -------------------------------------------------------------------------------- /test/nmm/40sqmm.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AzizZayed/DialEgg/HEAD/test/nmm/40sqmm.mlir -------------------------------------------------------------------------------- /test/nmm/4mm.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AzizZayed/DialEgg/HEAD/test/nmm/4mm.mlir -------------------------------------------------------------------------------- /test/nmm/5mm.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AzizZayed/DialEgg/HEAD/test/nmm/5mm.mlir -------------------------------------------------------------------------------- /test/nmm/80mm.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AzizZayed/DialEgg/HEAD/test/nmm/80mm.mlir -------------------------------------------------------------------------------- /test/nmm/80sqmm.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AzizZayed/DialEgg/HEAD/test/nmm/80sqmm.mlir -------------------------------------------------------------------------------- /test/nmm/nmm.egg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AzizZayed/DialEgg/HEAD/test/nmm/nmm.egg -------------------------------------------------------------------------------- /test/polynomial/polynomial.egg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AzizZayed/DialEgg/HEAD/test/polynomial/polynomial.egg -------------------------------------------------------------------------------- /test/polynomial/polynomial.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AzizZayed/DialEgg/HEAD/test/polynomial/polynomial.mlir -------------------------------------------------------------------------------- /test/run.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AzizZayed/DialEgg/HEAD/test/run.sh -------------------------------------------------------------------------------- /test/scf/scf_if.egg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AzizZayed/DialEgg/HEAD/test/scf/scf_if.egg -------------------------------------------------------------------------------- /test/scf/scf_if.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AzizZayed/DialEgg/HEAD/test/scf/scf_if.mlir -------------------------------------------------------------------------------- /test/stablehlo/tests.egg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AzizZayed/DialEgg/HEAD/test/stablehlo/tests.egg -------------------------------------------------------------------------------- /test/stablehlo/tests.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AzizZayed/DialEgg/HEAD/test/stablehlo/tests.mlir -------------------------------------------------------------------------------- /test/stablehlo_benchmarks/gmm.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AzizZayed/DialEgg/HEAD/test/stablehlo_benchmarks/gmm.csv -------------------------------------------------------------------------------- /test/stablehlo_benchmarks/gmm.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AzizZayed/DialEgg/HEAD/test/stablehlo_benchmarks/gmm.mlir -------------------------------------------------------------------------------- /test/stablehlo_benchmarks/jaxpt.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AzizZayed/DialEgg/HEAD/test/stablehlo_benchmarks/jaxpt.csv -------------------------------------------------------------------------------- /test/stablehlo_benchmarks/kex.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AzizZayed/DialEgg/HEAD/test/stablehlo_benchmarks/kex.csv -------------------------------------------------------------------------------- /test/stablehlo_benchmarks/kex.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AzizZayed/DialEgg/HEAD/test/stablehlo_benchmarks/kex.mlir -------------------------------------------------------------------------------- /test/stablehlo_benchmarks/llama.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AzizZayed/DialEgg/HEAD/test/stablehlo_benchmarks/llama.csv -------------------------------------------------------------------------------- /test/stablehlo_benchmarks/nasrnn.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AzizZayed/DialEgg/HEAD/test/stablehlo_benchmarks/nasrnn.csv -------------------------------------------------------------------------------- /test/stablehlo_benchmarks/nasrnn.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AzizZayed/DialEgg/HEAD/test/stablehlo_benchmarks/nasrnn.mlir -------------------------------------------------------------------------------- /test/stablehlo_benchmarks/resnet.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AzizZayed/DialEgg/HEAD/test/stablehlo_benchmarks/resnet.csv -------------------------------------------------------------------------------- /test/stablehlo_benchmarks/rewrites.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AzizZayed/DialEgg/HEAD/test/stablehlo_benchmarks/rewrites.txt -------------------------------------------------------------------------------- /test/stablehlo_benchmarks/stablehlo.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AzizZayed/DialEgg/HEAD/test/stablehlo_benchmarks/stablehlo.csv -------------------------------------------------------------------------------- /test/stablehlo_benchmarks/stablehlo.egg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AzizZayed/DialEgg/HEAD/test/stablehlo_benchmarks/stablehlo.egg -------------------------------------------------------------------------------- /test/stablehlo_benchmarks/stablehlo.min.egg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AzizZayed/DialEgg/HEAD/test/stablehlo_benchmarks/stablehlo.min.egg -------------------------------------------------------------------------------- /test/util/gen_libutil.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AzizZayed/DialEgg/HEAD/test/util/gen_libutil.sh -------------------------------------------------------------------------------- /test/util/util.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AzizZayed/DialEgg/HEAD/test/util/util.mlir -------------------------------------------------------------------------------- /test/vector_norm/vector_norm.egg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AzizZayed/DialEgg/HEAD/test/vector_norm/vector_norm.egg -------------------------------------------------------------------------------- /test/vector_norm/vector_norm.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AzizZayed/DialEgg/HEAD/test/vector_norm/vector_norm.mlir --------------------------------------------------------------------------------