Opaque relay op codegen

gussmith23 / glenside

A pure, low-level tensor program representation enabling tensor program optimization via program rewriting. See the web demo at https://gussmith23.github.io/glenside-web-demo/

71 stars 10 forks source link

Opaque relay op codegen #71

Closed hypercubestart closed 3 years ago

hypercubestart commented 4 years ago

TODO:

[x] implement codegen for relay ops
[x] implement add broadcasting
[x] add tests
- [x] RelayGlobalAvgPool2D
- [x] RelayBatchFlatten
- [x] RelayBiasAdd
[x] implement assertion to check result with out array (need to add functions on NDarray)
- [x] batchnorm
- [x] softmax
- [x] relu
- [x] maxpool2d

codegen for #65

hypercubestart commented 4 years ago

@gussmith23 build is failing because CI/CD machine doesn't have "math.h"

gussmith23 commented 4 years ago

Wait until #73 merged

gussmith23 commented 4 years ago

Interesting about math.h...odd that it wouldn't have it. Will debug later!