tensorgrad 0.0.14 → 0.0.16
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/dist/index.d.ts +154 -170
- package/dist/index.js +2208 -39
- package/dist/index.js.map +7 -1
- package/dist/worker.debug.js +553 -0
- package/package.json +60 -58
- package/src/adam.ts +69 -15
- package/src/compile.ts +334 -154
- package/src/index.ts +8 -4
- package/src/module.ts +72 -34
- package/src/runtime.ts +64 -11
- package/src/worker-protocol.ts +183 -0
- package/src/worker-proxy.ts +76 -0
- package/src/worker.ts +281 -0
- package/dist/adam.js +0 -111
- package/dist/adam.js.map +0 -1
- package/dist/buffers.js +0 -120
- package/dist/buffers.js.map +0 -1
- package/dist/capture.js +0 -33
- package/dist/capture.js.map +0 -1
- package/dist/codegen.js +0 -724
- package/dist/codegen.js.map +0 -1
- package/dist/compile.js +0 -180
- package/dist/compile.js.map +0 -1
- package/dist/grad.js +0 -380
- package/dist/grad.js.map +0 -1
- package/dist/ir.js +0 -60
- package/dist/ir.js.map +0 -1
- package/dist/module.js +0 -155
- package/dist/module.js.map +0 -1
- package/dist/nn.js +0 -135
- package/dist/nn.js.map +0 -1
- package/dist/ops.js +0 -326
- package/dist/ops.js.map +0 -1
- package/dist/runtime.js +0 -375
- package/dist/runtime.js.map +0 -1
- package/dist/shape.js +0 -259
- package/dist/shape.js.map +0 -1
- package/dist/trace.js +0 -100
- package/dist/trace.js.map +0 -1
package/dist/shape.js.map
DELETED
|
@@ -1 +0,0 @@
|
|
|
1
|
-
{"version":3,"file":"shape.js","sourceRoot":"","sources":["../src/shape.ts"],"names":[],"mappings":"AAAA,mDAAmD;AACnD,EAAE;AACF,4EAA4E;AAC5E,wEAAwE;AACxE,+DAA+D;AAC/D,EAAE;AACF,6CAA6C;AAC7C,0EAA0E;AAC1E,wEAAwE;AACxE,uEAAuE;AACvE,+CAA+C;AAC/C,iDAAiD;AACjD,oDAAoD;AACpD,+DAA+D;AAC/D,qFAAqF;AACrF,6EAA6E;AAC7E,+EAA+E;AAG/E,OAAO,EAAE,UAAU,EAAE,MAAM,SAAS,CAAA;AAEpC,+EAA+E;AAC/E,SAAS;AACT,+EAA+E;AAE/E,MAAM,OAAO,UAAW,SAAQ,KAAK;IACnC,YAAY,OAAe,EAAE,IAAqB;QAChD,MAAM,SAAS,GAAG,IAAI,CAAC,CAAC,CAAC,GAAG,OAAO,UAAU,UAAU,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC,CAAC,OAAO,CAAA;QACzE,KAAK,CAAC,SAAS,CAAC,CAAA;QAChB,IAAI,CAAC,IAAI,GAAG,YAAY,CAAA;IAC1B,CAAC;CACF;AAED,SAAS,IAAI,CAAC,OAAe,EAAE,IAAqB;IAClD,MAAM,IAAI,UAAU,CAAC,OAAO,EAAE,IAAI,CAAC,CAAA;AACrC,CAAC;AAED,+EAA+E;AAC/E,kBAAkB;AAClB,+EAA+E;AAE/E,MAAM,UAAU,WAAW,CAAC,CAAQ,EAAE,CAAQ;IAC5C,IAAI,CAAC,CAAC,MAAM,KAAK,CAAC,CAAC,MAAM;QAAE,OAAO,KAAK,CAAA;IACvC,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,CAAC,CAAC,MAAM,EAAE,CAAC,EAAE;QAAE,IAAI,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;YAAE,OAAO,KAAK,CAAA;IAClE,OAAO,IAAI,CAAA;AACb,CAAC;AAED,MAAM,UAAU,SAAS,CAAC,KAAY;IACpC,IAAI,CAAC,GAAG,CAAC,CAAA;IACT,KAAK,MAAM,CAAC,IAAI,KAAK;QAAE,CAAC,IAAI,CAAC,CAAA;IAC7B,OAAO,CAAC,CAAA;AACV,CAAC;AAED,MAAM,UAAU,SAAS,CAAC,KAAY;IACpC,OAAO,IAAI,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,GAAG,CAAA;AAChC,CAAC;AAED,8EAA8E;AAC9E,+EAA+E;AAC/E,qEAAqE;AACrE,MAAM,UAAU,iBAAiB,CAAC,CAAQ,EAAE,CAAQ;IAClD,MAAM,IAAI,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC,CAAC,MAAM,CAAC,CAAA;IACzC,MAAM,GAAG,GAAa,IAAI,KAAK,CAAC,IAAI,CAAC,CAAA;IACrC,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,IAAI,EAAE,CAAC,EAAE,EAAE,CAAC;QAC9B,MAAM,EAAE,GAAG,CAAC,GAAG,CAAC,IAAI,GAAG,CAAC,CAAC,MAAM,CAAC,CAAA;QAChC,MAAM,EAAE,GAAG,CAAC,GAAG,CAAC,IAAI,GAAG,CAAC,CAAC,MAAM,CAAC,CAAA;QAChC,MAAM,EAAE,GAAG,EAAE,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,EAAE,CAAE,CAAA;QAC9B,MAAM,EAAE,GAAG,EAAE,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,EAAE,CAAE,CAAA;QAC9B,IAAI,EAAE,KAAK,EAAE;YAAE,GAAG,CAAC,CAAC,CAAC,GAAG,EAAE,CAAA;aACrB,IAAI,EAAE,KAAK,CAAC;YAAE,GAAG,CAAC,CAAC,CAAC,GAAG,EAAE,CAAA;aACzB,IAAI,EAAE,KAAK,CAAC;YAAE,GAAG,CAAC,CAAC,CAAC,GAAG,EAAE,CAAA;;YACzB,OAAO,IAAI,CAAA;IAClB,CAAC;IACD,OAAO,GAAG,CAAA;AACZ,CAAC;AAED,+EAA+E;AAC/E,qBAAqB;AACrB,+EAA+E;AAC/E,EAAE;AACF,4EAA4E;AAC5E,mDAAmD;AAEnD,MAAM,UAAU,qBAAqB,CACnC,MAAc,EAAE,MAAa,EAAE,MAAa,EAAE,IAAqB;IAEnE,MAAM,MAAM,GAAG,iBAAiB,CAAC,MAAM,EAAE,MAAM,CAAC,CAAA;IAChD,IAAI,CAAC,MAAM,EAAE,CAAC;QACZ,IAAI,CACF,GAAG,MAAM,yBAAyB,SAAS,CAAC,MAAM,CAAC,QAAQ,SAAS,CAAC,MAAM,CAAC,IAAI;YAChF,wFAAwF;YACxF,4CAA4C,EAC5C,IAAI,CACL,CAAA;IACH,CAAC;IACD,OAAO,MAAM,CAAA;AACf,CAAC;AAED,MAAM,UAAU,UAAU,CAAC,OAAe,EAAE,MAAa,EAAE,KAAsB;IAC/E,OAAO,MAAM,CAAA;AACf,CAAC;AAED,MAAM,UAAU,aAAa,CAAC,MAAc,EAAE,MAAa,EAAE,IAAqB;IAChF,IAAI,MAAM,CAAC,MAAM,KAAK,CAAC;QAAE,IAAI,CAAC,GAAG,MAAM,8BAA8B,EAAE,IAAI,CAAC,CAAA;IAC5E,2CAA2C;IAC3C,OAAO,CAAC,GAAG,MAAM,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAA;AACpC,CAAC;AAED,MAAM,UAAU,YAAY,CAAC,MAAc,EAAE,MAAa,EAAE,IAAqB;IAC/E,IAAI,MAAM,CAAC,MAAM,KAAK,CAAC;QAAE,IAAI,CAAC,GAAG,MAAM,8BAA8B,EAAE,IAAI,CAAC,CAAA;IAC5E,sCAAsC;IACtC,OAAO,MAAM,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAA;AAC5B,CAAC;AAED,MAAM,UAAU,YAAY,CAAC,MAAc,EAAE,MAAa,EAAE,QAAe,EAAE,IAAqB;IAChG,sEAAsE;IACtE,IAAI,QAAQ,GAAG,CAAC,CAAC,CAAA;IACjB,IAAI,SAAS,GAAG,CAAC,CAAA;IACjB,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,QAAQ,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;QACzC,MAAM,CAAC,GAAG,QAAQ,CAAC,CAAC,CAAE,CAAA;QACtB,IAAI,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC;YACb,IAAI,QAAQ,KAAK,CAAC,CAAC;gBAAE,IAAI,CAAC,GAAG,MAAM,4CAA4C,SAAS,CAAC,QAAQ,CAAC,EAAE,EAAE,IAAI,CAAC,CAAA;YAC3G,QAAQ,GAAG,CAAC,CAAA;QACd,CAAC;aAAM,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC;YAClB,IAAI,CAAC,GAAG,MAAM,iBAAiB,CAAC,gBAAgB,SAAS,CAAC,QAAQ,CAAC,EAAE,EAAE,IAAI,CAAC,CAAA;QAC9E,CAAC;aAAM,CAAC;YACN,SAAS,IAAI,CAAC,CAAA;QAChB,CAAC;IACH,CAAC;IACD,MAAM,OAAO,GAAG,SAAS,CAAC,MAAM,CAAC,CAAA;IACjC,MAAM,GAAG,GAAG,CAAC,GAAG,QAAQ,CAAC,CAAA;IACzB,IAAI,QAAQ,KAAK,CAAC,CAAC,EAAE,CAAC;QACpB,IAAI,OAAO,GAAG,SAAS,KAAK,CAAC,EAAE,CAAC;YAC9B,IAAI,CAAC,GAAG,MAAM,oBAAoB,SAAS,CAAC,MAAM,CAAC,UAAU,OAAO,QAAQ,SAAS,CAAC,QAAQ,CAAC,6BAA6B,SAAS,EAAE,EAAE,IAAI,CAAC,CAAA;QAChJ,CAAC;QACD,GAAG,CAAC,QAAQ,CAAC,GAAG,OAAO,GAAG,SAAS,CAAA;IACrC,CAAC;SAAM,IAAI,SAAS,KAAK,OAAO,EAAE,CAAC;QACjC,IAAI,CAAC,GAAG,MAAM,2BAA2B,SAAS,CAAC,MAAM,CAAC,QAAQ,OAAO,0BAA0B,SAAS,CAAC,QAAQ,CAAC,QAAQ,SAAS,EAAE,EAAE,IAAI,CAAC,CAAA;IAClJ,CAAC;IACD,OAAO,GAAG,CAAA;AACZ,CAAC;AAED,MAAM,UAAU,cAAc,CAAC,MAAc,EAAE,MAAa,EAAE,IAAuB,EAAE,IAAqB;IAC1G,IAAI,IAAI,CAAC,MAAM,KAAK,MAAM,CAAC,MAAM,EAAE,CAAC;QAClC,IAAI,CAAC,GAAG,MAAM,iBAAiB,IAAI,CAAC,MAAM,0BAA0B,MAAM,CAAC,MAAM,EAAE,EAAE,IAAI,CAAC,CAAA;IAC5F,CAAC;IACD,MAAM,IAAI,GAAG,IAAI,GAAG,EAAU,CAAA;IAC9B,KAAK,MAAM,CAAC,IAAI,IAAI,EAAE,CAAC;QACrB,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,IAAI,MAAM,CAAC,MAAM;YAAE,IAAI,CAAC,GAAG,MAAM,gBAAgB,CAAC,0BAA0B,MAAM,CAAC,MAAM,EAAE,EAAE,IAAI,CAAC,CAAA;QAChH,IAAI,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC;YAAE,IAAI,CAAC,GAAG,MAAM,8BAA8B,CAAC,EAAE,EAAE,IAAI,CAAC,CAAA;QACvE,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,CAAA;IACb,CAAC;IACD,OAAO,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,MAAM,CAAC,CAAC,CAAE,CAAC,CAAA;AAClC,CAAC;AAED,qEAAqE;AACrE,MAAM,UAAU,WAAW,CAAC,MAAc,EAAE,MAAa,EAAE,MAAa,EAAE,IAAqB;IAC7F,IAAI,MAAM,CAAC,MAAM,GAAG,CAAC;QAAE,IAAI,CAAC,GAAG,MAAM,kCAAkC,SAAS,CAAC,MAAM,CAAC,EAAE,EAAE,IAAI,CAAC,CAAA;IACjG,IAAI,MAAM,CAAC,MAAM,KAAK,CAAC;QAAE,IAAI,CAAC,GAAG,MAAM,+BAA+B,SAAS,CAAC,MAAM,CAAC,sCAAsC,EAAE,IAAI,CAAC,CAAA;IACpI,MAAM,CAAC,GAAG,MAAM,CAAC,MAAM,CAAC,MAAM,GAAG,CAAC,CAAE,CAAA;IACpC,MAAM,EAAE,GAAG,MAAM,CAAC,MAAM,CAAC,MAAM,GAAG,CAAC,CAAE,CAAA;IACrC,MAAM,EAAE,GAAG,MAAM,CAAC,CAAC,CAAE,CAAA;IACrB,MAAM,CAAC,GAAG,MAAM,CAAC,CAAC,CAAE,CAAA;IACpB,IAAI,EAAE,KAAK,EAAE;QAAE,IAAI,CAAC,GAAG,MAAM,8BAA8B,SAAS,CAAC,MAAM,CAAC,MAAM,SAAS,CAAC,MAAM,CAAC,wBAAwB,EAAE,yBAAyB,EAAE,GAAG,EAAE,IAAI,CAAC,CAAA;IAClK,OAAO,CAAC,GAAG,MAAM,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC,CAAA;AACvC,CAAC;AAED,gGAAgG;AAChG,MAAM,UAAU,kBAAkB,CAAC,MAAc,EAAE,MAAa,EAAE,MAAa,EAAE,IAAqB;IACpG,IAAI,MAAM,CAAC,MAAM,GAAG,CAAC,IAAI,MAAM,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QAC3C,IAAI,CAAC,GAAG,MAAM,0CAA0C,SAAS,CAAC,MAAM,CAAC,QAAQ,SAAS,CAAC,MAAM,CAAC,EAAE,EAAE,IAAI,CAAC,CAAA;IAC7G,CAAC;IACD,IAAI,MAAM,CAAC,MAAM,KAAK,MAAM,CAAC,MAAM,EAAE,CAAC;QACpC,IAAI,CAAC,GAAG,MAAM,2BAA2B,MAAM,CAAC,MAAM,OAAO,MAAM,CAAC,MAAM,8CAA8C,EAAE,IAAI,CAAC,CAAA;IACjI,CAAC;IACD,MAAM,MAAM,GAAG,MAAM,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAA;IAClC,MAAM,MAAM,GAAG,MAAM,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAA;IAClC,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,MAAM,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;QACvC,IAAI,MAAM,CAAC,CAAC,CAAC,KAAK,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC;YAC5B,IAAI,CAAC,GAAG,MAAM,6BAA6B,SAAS,CAAC,MAAM,CAAC,OAAO,SAAS,CAAC,MAAM,CAAC,EAAE,EAAE,IAAI,CAAC,CAAA;QAC/F,CAAC;IACH,CAAC;IACD,MAAM,CAAC,GAAG,MAAM,CAAC,MAAM,CAAC,MAAM,GAAG,CAAC,CAAE,CAAA;IACpC,MAAM,EAAE,GAAG,MAAM,CAAC,MAAM,CAAC,MAAM,GAAG,CAAC,CAAE,CAAA;IACrC,MAAM,EAAE,GAAG,MAAM,CAAC,MAAM,CAAC,MAAM,GAAG,CAAC,CAAE,CAAA;IACrC,MAAM,CAAC,GAAG,MAAM,CAAC,MAAM,CAAC,MAAM,GAAG,CAAC,CAAE,CAAA;IACpC,IAAI,EAAE,KAAK,EAAE;QAAE,IAAI,CAAC,GAAG,MAAM,iDAAiD,EAAE,6BAA6B,EAAE,EAAE,EAAE,IAAI,CAAC,CAAA;IACxH,OAAO,CAAC,GAAG,MAAM,EAAE,CAAC,EAAE,CAAC,CAAC,CAAA;AAC1B,CAAC;AAED,MAAM,UAAU,WAAW,CAAC,MAAc,EAAE,YAAmB,EAAE,KAAa,EAAE,IAAqB;IACnG,IAAI,KAAK,IAAI,CAAC;QAAE,IAAI,CAAC,GAAG,MAAM,iCAAiC,KAAK,EAAE,EAAE,IAAI,CAAC,CAAA;IAC7E,OAAO,CAAC,GAAG,YAAY,EAAE,KAAK,CAAC,CAAA;AACjC,CAAC;AAED,4EAA4E;AAC5E,MAAM,UAAU,gBAAgB,CAAC,MAAc,EAAE,MAAa,EAAE,IAAqB;IACnF,IAAI,MAAM,CAAC,MAAM,GAAG,CAAC;QAAE,IAAI,CAAC,GAAG,MAAM,6BAA6B,SAAS,CAAC,MAAM,CAAC,EAAE,EAAE,IAAI,CAAC,CAAA;IAC5F,MAAM,CAAC,GAAG,MAAM,CAAC,MAAM,CAAC,MAAM,GAAG,CAAC,CAAE,CAAA;IACpC,MAAM,CAAC,GAAG,MAAM,CAAC,MAAM,CAAC,MAAM,GAAG,CAAC,CAAE,CAAA;IACpC,IAAI,CAAC,KAAK,CAAC;QAAE,IAAI,CAAC,GAAG,MAAM,oDAAoD,SAAS,CAAC,MAAM,CAAC,EAAE,EAAE,IAAI,CAAC,CAAA;IACzG,OAAO,MAAM,CAAA;AACf,CAAC;AAED,MAAM,UAAU,mBAAmB,CAAC,MAAc,EAAE,MAAa,EAAE,KAAa,EAAE,GAAW,EAAE,IAAqB;IAClH,IAAI,MAAM,CAAC,MAAM,KAAK,CAAC;QAAE,IAAI,CAAC,GAAG,MAAM,2BAA2B,EAAE,IAAI,CAAC,CAAA;IACzE,MAAM,IAAI,GAAG,MAAM,CAAC,MAAM,CAAC,MAAM,GAAG,CAAC,CAAE,CAAA;IACvC,IAAI,KAAK,GAAG,CAAC,IAAI,GAAG,GAAG,IAAI,IAAI,KAAK,IAAI,GAAG,EAAE,CAAC;QAC5C,IAAI,CAAC,GAAG,MAAM,oBAAoB,KAAK,KAAK,GAAG,2BAA2B,IAAI,EAAE,EAAE,IAAI,CAAC,CAAA;IACzF,CAAC;IACD,OAAO,CAAC,GAAG,MAAM,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,GAAG,GAAG,KAAK,CAAC,CAAA;AAC9C,CAAC;AAED,4EAA4E;AAC5E,6DAA6D;AAC7D,MAAM,UAAU,gBAAgB,CAAC,MAAc,EAAE,MAAa,EAAE,WAAkB,EAAE,IAAqB;IACvG,IAAI,MAAM,CAAC,MAAM,GAAG,WAAW,CAAC,MAAM,EAAE,CAAC;QACvC,IAAI,CAAC,GAAG,MAAM,iBAAiB,MAAM,CAAC,MAAM,kBAAkB,WAAW,CAAC,MAAM,EAAE,EAAE,IAAI,CAAC,CAAA;IAC3F,CAAC;IACD,MAAM,MAAM,GAAG,WAAW,CAAC,MAAM,GAAG,MAAM,CAAC,MAAM,CAAA;IACjD,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,MAAM,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;QACvC,MAAM,EAAE,GAAG,MAAM,CAAC,CAAC,CAAE,CAAA;QACrB,MAAM,EAAE,GAAG,WAAW,CAAC,MAAM,GAAG,CAAC,CAAE,CAAA;QACnC,IAAI,EAAE,KAAK,EAAE,IAAI,EAAE,KAAK,CAAC,EAAE,CAAC;YAC1B,IAAI,CAAC,GAAG,MAAM,sBAAsB,SAAS,CAAC,MAAM,CAAC,OAAO,SAAS,CAAC,WAAW,CAAC,WAAW,CAAC,UAAU,EAAE,+BAA+B,MAAM,GAAG,CAAC,UAAU,EAAE,eAAe,EAAE,IAAI,CAAC,CAAA;QACvL,CAAC;IACH,CAAC;IACD,OAAO,WAAW,CAAA;AACpB,CAAC;AAED,+EAA+E;AAC/E,iFAAiF;AACjF,MAAM,UAAU,eAAe,CAAC,MAAc,EAAE,MAAa,EAAE,WAAkB,EAAE,IAAqB;IACtG,IAAI,WAAW,CAAC,MAAM,GAAG,MAAM,CAAC,MAAM,EAAE,CAAC;QACvC,IAAI,CAAC,GAAG,MAAM,iBAAiB,WAAW,CAAC,MAAM,kBAAkB,MAAM,CAAC,MAAM,EAAE,EAAE,IAAI,CAAC,CAAA;IAC3F,CAAC;IACD,MAAM,MAAM,GAAG,MAAM,CAAC,MAAM,GAAG,WAAW,CAAC,MAAM,CAAA;IACjD,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,WAAW,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;QAC5C,MAAM,EAAE,GAAG,MAAM,CAAC,MAAM,GAAG,CAAC,CAAE,CAAA;QAC9B,MAAM,EAAE,GAAG,WAAW,CAAC,CAAC,CAAE,CAAA;QAC1B,IAAI,EAAE,KAAK,EAAE,IAAI,EAAE,KAAK,CAAC,EAAE,CAAC;YAC1B,IAAI,CAAC,GAAG,MAAM,uBAAuB,SAAS,CAAC,MAAM,CAAC,OAAO,SAAS,CAAC,WAAW,CAAC,kBAAkB,CAAC,UAAU,EAAE,6BAA6B,EAAE,IAAI,CAAC,CAAA;QACxJ,CAAC;IACH,CAAC;IACD,OAAO,WAAW,CAAA;AACpB,CAAC;AAED,+EAA+E;AAC/E,gDAAgD;AAChD,MAAM,UAAU,UAAU,CAAC,MAAc,EAAE,SAAgB,EAAE,MAAa,EAAE,MAAa,EAAE,IAAqB;IAC9G,MAAM,EAAE,GAAG,iBAAiB,CAAC,MAAM,EAAE,MAAM,CAAC,CAAA;IAC5C,IAAI,CAAC,EAAE;QAAE,IAAI,CAAC,GAAG,MAAM,uBAAuB,SAAS,CAAC,MAAM,CAAC,OAAO,SAAS,CAAC,MAAM,CAAC,EAAE,EAAE,IAAI,CAAC,CAAA;IAChG,MAAM,MAAM,GAAG,iBAAiB,CAAC,SAAS,EAAE,EAAE,CAAC,CAAA;IAC/C,IAAI,CAAC,MAAM;QAAE,IAAI,CAAC,GAAG,MAAM,UAAU,SAAS,CAAC,SAAS,CAAC,sCAAsC,SAAS,CAAC,EAAE,CAAC,EAAE,EAAE,IAAI,CAAC,CAAA;IACrH,OAAO,MAAM,CAAA;AACf,CAAC;AAED,MAAM,UAAU,aAAa,CAAC,MAAc,EAAE,MAAa,EAAE,OAAc,EAAE,IAAqB;IAChG,IAAI,CAAC,WAAW,CAAC,MAAM,EAAE,OAAO,CAAC,EAAE,CAAC;QAClC,IAAI,CAAC,GAAG,MAAM,6CAA6C,SAAS,CAAC,MAAM,CAAC,QAAQ,SAAS,CAAC,OAAO,CAAC,EAAE,EAAE,IAAI,CAAC,CAAA;IACjH,CAAC;IACD,OAAO,MAAM,CAAA;AACf,CAAC"}
|
package/dist/trace.js
DELETED
|
@@ -1,100 +0,0 @@
|
|
|
1
|
-
// Trace driver. Holds the "current graph" in module-local state so user code
|
|
2
|
-
// can call ops without threading a graph parameter through every function.
|
|
3
|
-
//
|
|
4
|
-
// Usage:
|
|
5
|
-
//
|
|
6
|
-
// const graph = trace(() => {
|
|
7
|
-
// const x = tensorInput('x', [B, T], 'i32')
|
|
8
|
-
// const w = paramInput('w', [V, D], 'f32')
|
|
9
|
-
// // ... user computation building tensors ...
|
|
10
|
-
// return finalLossTensor
|
|
11
|
-
// })
|
|
12
|
-
//
|
|
13
|
-
// `trace` is single-threaded and re-entrant only via nested calls (which share
|
|
14
|
-
// the outer graph — but we don't currently have a use for nesting). Calling an
|
|
15
|
-
// op outside a `trace(...)` block is an error.
|
|
16
|
-
import { makeGraph, addOp, captureSite } from './ir.js';
|
|
17
|
-
// Module-local: the graph being built right now, or null if no trace is active.
|
|
18
|
-
let _current = null;
|
|
19
|
-
// Module-local: whether `capture(name, t)` calls should register on the current
|
|
20
|
-
// graph. True only during the user's forward trace; false during `traceInto`
|
|
21
|
-
// (autograd / optimizer ops shouldn't accidentally publish gradient tensors).
|
|
22
|
-
let _captureEnabled = false;
|
|
23
|
-
export function currentGraph() {
|
|
24
|
-
if (!_current) {
|
|
25
|
-
throw new Error('tensorgrad: ops can only be called inside trace(). ' +
|
|
26
|
-
'Did you forget to wrap your forward pass?');
|
|
27
|
-
}
|
|
28
|
-
return _current;
|
|
29
|
-
}
|
|
30
|
-
export function isCaptureEnabled() {
|
|
31
|
-
return _captureEnabled;
|
|
32
|
-
}
|
|
33
|
-
// Run `fn` with a fresh graph as the current one; capture and return the graph.
|
|
34
|
-
// `fn` must return the tensor (or array of tensors) to mark as graph outputs.
|
|
35
|
-
export function trace(fn) {
|
|
36
|
-
if (_current) {
|
|
37
|
-
throw new Error('tensorgrad: nested trace() is not supported');
|
|
38
|
-
}
|
|
39
|
-
const g = makeGraph();
|
|
40
|
-
_current = g;
|
|
41
|
-
_captureEnabled = true;
|
|
42
|
-
try {
|
|
43
|
-
const result = fn();
|
|
44
|
-
const outputs = Array.isArray(result) ? result : [result];
|
|
45
|
-
for (const t of outputs) {
|
|
46
|
-
;
|
|
47
|
-
g.outputs.push(t.id);
|
|
48
|
-
}
|
|
49
|
-
}
|
|
50
|
-
finally {
|
|
51
|
-
_current = null;
|
|
52
|
-
_captureEnabled = false;
|
|
53
|
-
}
|
|
54
|
-
return g;
|
|
55
|
-
}
|
|
56
|
-
// Re-enter an existing graph to append more ops. Used by autograd to add
|
|
57
|
-
// backward ops to a graph that's already been traced. `fn` runs with the
|
|
58
|
-
// supplied graph as the current one; any ops it calls append to that graph.
|
|
59
|
-
// Capture is intentionally disabled here — backward / optimizer rules
|
|
60
|
-
// shouldn't publish their internal tensors via `capture()`.
|
|
61
|
-
// Returns whatever `fn` returns.
|
|
62
|
-
export function traceInto(g, fn) {
|
|
63
|
-
if (_current) {
|
|
64
|
-
throw new Error('tensorgrad: traceInto() called while another trace is active');
|
|
65
|
-
}
|
|
66
|
-
_current = g;
|
|
67
|
-
// _captureEnabled stays false (default) — explicit, but not toggled.
|
|
68
|
-
try {
|
|
69
|
-
return fn();
|
|
70
|
-
}
|
|
71
|
-
finally {
|
|
72
|
-
_current = null;
|
|
73
|
-
}
|
|
74
|
-
}
|
|
75
|
-
function assertNameUnused(g, name, kinds, label) {
|
|
76
|
-
if (g.ops.some(op => kinds.includes(op.kind) && op.name === name)) {
|
|
77
|
-
throw new Error(`tensorgrad: ${label} name '${name}' already used in this trace`);
|
|
78
|
-
}
|
|
79
|
-
}
|
|
80
|
-
export function paramInput(name, shape, dtype = 'f32') {
|
|
81
|
-
const g = currentGraph();
|
|
82
|
-
assertNameUnused(g, name, ['param_input', 'tensor_input'], 'input');
|
|
83
|
-
const site = captureSite('paramInput');
|
|
84
|
-
return addOp(g, 'param_input', shape, dtype, site, { name });
|
|
85
|
-
}
|
|
86
|
-
export function tensorInput(name, shape, dtype = 'f32') {
|
|
87
|
-
const g = currentGraph();
|
|
88
|
-
assertNameUnused(g, name, ['param_input', 'tensor_input'], 'input');
|
|
89
|
-
const site = captureSite('tensorInput');
|
|
90
|
-
return addOp(g, 'tensor_input', shape, dtype, site, { name });
|
|
91
|
-
}
|
|
92
|
-
// Persistent state buffer. Allocated at compile time, zero-(or initValue-)initialized,
|
|
93
|
-
// and updated across step() calls via writebacks declared by the optimizer helper.
|
|
94
|
-
export function stateInput(name, shape, dtype = 'f32', initValue = 0) {
|
|
95
|
-
const g = currentGraph();
|
|
96
|
-
assertNameUnused(g, name, ['state_input'], 'state');
|
|
97
|
-
const site = captureSite('stateInput');
|
|
98
|
-
return addOp(g, 'state_input', shape, dtype, site, { name, initValue });
|
|
99
|
-
}
|
|
100
|
-
//# sourceMappingURL=trace.js.map
|
package/dist/trace.js.map
DELETED
|
@@ -1 +0,0 @@
|
|
|
1
|
-
{"version":3,"file":"trace.js","sourceRoot":"","sources":["../src/trace.ts"],"names":[],"mappings":"AAAA,6EAA6E;AAC7E,2EAA2E;AAC3E,EAAE;AACF,SAAS;AACT,EAAE;AACF,gCAAgC;AAChC,gDAAgD;AAChD,+CAA+C;AAC/C,mDAAmD;AACnD,6BAA6B;AAC7B,OAAO;AACP,EAAE;AACF,+EAA+E;AAC/E,+EAA+E;AAC/E,+CAA+C;AAG/C,OAAO,EAAE,SAAS,EAAE,KAAK,EAAE,WAAW,EAAE,MAAM,SAAS,CAAA;AAEvD,gFAAgF;AAChF,IAAI,QAAQ,GAAiB,IAAI,CAAA;AACjC,gFAAgF;AAChF,6EAA6E;AAC7E,8EAA8E;AAC9E,IAAI,eAAe,GAAG,KAAK,CAAA;AAE3B,MAAM,UAAU,YAAY;IAC1B,IAAI,CAAC,QAAQ,EAAE,CAAC;QACd,MAAM,IAAI,KAAK,CACb,qDAAqD;YACrD,2CAA2C,CAC5C,CAAA;IACH,CAAC;IACD,OAAO,QAAQ,CAAA;AACjB,CAAC;AAED,MAAM,UAAU,gBAAgB;IAC9B,OAAO,eAAe,CAAA;AACxB,CAAC;AAED,gFAAgF;AAChF,8EAA8E;AAC9E,MAAM,UAAU,KAAK,CAAC,EAA2B;IAC/C,IAAI,QAAQ,EAAE,CAAC;QACb,MAAM,IAAI,KAAK,CAAC,6CAA6C,CAAC,CAAA;IAChE,CAAC;IACD,MAAM,CAAC,GAAG,SAAS,EAAE,CAAA;IACrB,QAAQ,GAAG,CAAC,CAAA;IACZ,eAAe,GAAG,IAAI,CAAA;IACtB,IAAI,CAAC;QACH,MAAM,MAAM,GAAG,EAAE,EAAE,CAAA;QACnB,MAAM,OAAO,GAAG,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,CAAA;QACzD,KAAK,MAAM,CAAC,IAAI,OAAO,EAAE,CAAC;YACxB,CAAC;YAAC,CAAC,CAAC,OAAoB,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,CAAA;QACrC,CAAC;IACH,CAAC;YAAS,CAAC;QACT,QAAQ,GAAG,IAAI,CAAA;QACf,eAAe,GAAG,KAAK,CAAA;IACzB,CAAC;IACD,OAAO,CAAC,CAAA;AACV,CAAC;AAED,yEAAyE;AACzE,yEAAyE;AACzE,4EAA4E;AAC5E,sEAAsE;AACtE,4DAA4D;AAC5D,iCAAiC;AACjC,MAAM,UAAU,SAAS,CAAI,CAAQ,EAAE,EAAW;IAChD,IAAI,QAAQ,EAAE,CAAC;QACb,MAAM,IAAI,KAAK,CAAC,8DAA8D,CAAC,CAAA;IACjF,CAAC;IACD,QAAQ,GAAG,CAAC,CAAA;IACZ,qEAAqE;IACrE,IAAI,CAAC;QACH,OAAO,EAAE,EAAE,CAAA;IACb,CAAC;YAAS,CAAC;QACT,QAAQ,GAAG,IAAI,CAAA;IACjB,CAAC;AACH,CAAC;AAUD,SAAS,gBAAgB,CAAC,CAAQ,EAAE,IAAY,EAAE,KAAuB,EAAE,KAAa;IACtF,IAAI,CAAC,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC,EAAE,CAAC,KAAK,CAAC,QAAQ,CAAC,EAAE,CAAC,IAAsB,CAAC,IAAK,EAAwB,CAAC,IAAI,KAAK,IAAI,CAAC,EAAE,CAAC;QAC3G,MAAM,IAAI,KAAK,CAAC,eAAe,KAAK,UAAU,IAAI,8BAA8B,CAAC,CAAA;IACnF,CAAC;AACH,CAAC;AAED,MAAM,UAAU,UAAU,CAAC,IAAY,EAAE,KAAY,EAAE,QAAe,KAAK;IACzE,MAAM,CAAC,GAAG,YAAY,EAAE,CAAA;IACxB,gBAAgB,CAAC,CAAC,EAAE,IAAI,EAAE,CAAC,aAAa,EAAE,cAAc,CAAC,EAAE,OAAO,CAAC,CAAA;IACnE,MAAM,IAAI,GAAG,WAAW,CAAC,YAAY,CAAC,CAAA;IACtC,OAAO,KAAK,CAAC,CAAC,EAAE,aAAa,EAAE,KAAK,EAAE,KAAK,EAAE,IAAI,EAAE,EAAE,IAAI,EAAS,CAAC,CAAA;AACrE,CAAC;AAED,MAAM,UAAU,WAAW,CAAC,IAAY,EAAE,KAAY,EAAE,QAAe,KAAK;IAC1E,MAAM,CAAC,GAAG,YAAY,EAAE,CAAA;IACxB,gBAAgB,CAAC,CAAC,EAAE,IAAI,EAAE,CAAC,aAAa,EAAE,cAAc,CAAC,EAAE,OAAO,CAAC,CAAA;IACnE,MAAM,IAAI,GAAG,WAAW,CAAC,aAAa,CAAC,CAAA;IACvC,OAAO,KAAK,CAAC,CAAC,EAAE,cAAc,EAAE,KAAK,EAAE,KAAK,EAAE,IAAI,EAAE,EAAE,IAAI,EAAS,CAAC,CAAA;AACtE,CAAC;AAED,uFAAuF;AACvF,mFAAmF;AACnF,MAAM,UAAU,UAAU,CAAC,IAAY,EAAE,KAAY,EAAE,QAAe,KAAK,EAAE,SAAS,GAAG,CAAC;IACxF,MAAM,CAAC,GAAG,YAAY,EAAE,CAAA;IACxB,gBAAgB,CAAC,CAAC,EAAE,IAAI,EAAE,CAAC,aAAa,CAAC,EAAE,OAAO,CAAC,CAAA;IACnD,MAAM,IAAI,GAAG,WAAW,CAAC,YAAY,CAAC,CAAA;IACtC,OAAO,KAAK,CAAC,CAAC,EAAE,aAAa,EAAE,KAAK,EAAE,KAAK,EAAE,IAAI,EAAE,EAAE,IAAI,EAAE,SAAS,EAAS,CAAC,CAAA;AAChF,CAAC"}
|