Skip to content

stelleg/tlc

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

8 Commits
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Tensor Language Compiler

Compiles a nameless lambda interpretation of einstein notation as a Haskell EDSL into LLVM IR.

examples :: [Function] 
examples = [
 Function "vsmul" [(a, ft []), (b, ft [d])] $
  Λ d i $ v a * b!i, 
 Function "vvmul" [(a, ft [d]), (b, ft [d])] $
  Σ i $ a!i * b!i,
 Function "mvmul" [(a, ft [d, d]), (x, ft [d])] $
  Λ d j (Σ i (a!j!i * x!i)),
 Function "transpose" [(a, ft [d, d])] $
  Λ d i $ Λ d j $ a!j!i,
 Function "t3v" [(b, ft [d,d,d]), (c, ft [d])] $ 
  Λ d i $ Λ d j $ Σ k $ b!i!j!k * c!k
  ]

Sadly this was implemented in the good old days of typed pointers in LLVM, so you'll need to compile against such an oldie.

About

tensor language compiler

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published