Knossos
Contents:
Welcome to Knossos!
Compilation
Derivatives
So what? I can do that in PyTorch
Integrating with PyTorch
Limitations
Examples
Relu3 - An example kernel
Relu3 MNIST - Using Rrelu3 in a PyTorch model
Sqrl - An example tensor-to-tensor kernel
Knossos IR
Benchmarking
MLIR integration
About Knossos
Knossos Glossary
Knossos
»
Examples
»
Relu3 MNIST - Using Rrelu3 in a PyTorch model
View page source
Relu3 MNIST - Using Rrelu3 in a PyTorch model
ΒΆ