03be33eea4
Relax the requirements on CustomOp. ( #486 )
...
* Relax the requirements on CustomOp.
* Simplify the custom-ops when no backward is required.
2023-08-17 11:12:05 +01:00
1235aa2536
Use bail rather than wrapping a string where possible. ( #249 )
...
* Use bail rather than wrapping a string where possible.
* Revert the cuda default bit.
2023-07-26 15:42:46 +01:00
1a5416ec35
Rename exposed ops.
2023-07-26 12:43:19 +02:00
b6f7dfb682
CPU implementation for the custom RMS example. ( #228 )
...
* CPU implementation for the custom RMS example.
* Add the eps parameter.
2023-07-23 20:04:20 +01:00
e449ce53a2
Wrapping code to call the custom op. ( #225 )
...
* Wrapping code to call the custom op.
* Get the rms example to work.
* Get around rustfmt failing in the CI.
* Fix the rms computation.
2023-07-23 11:31:17 +01:00
b8a10425ad
Kernel build example ( #224 )
...
* Build example kernels.
* Add some sample custom kernel.
* Get the example kernel to compile.
* Add some cuda code.
* More cuda custom op.
* More cuda custom ops.
2023-07-23 07:15:37 +01:00
