diff options
| author | Laurent Mazare <laurent.mazare@gmail.com> | 2023-07-23 08:15:37 +0200 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2023-07-23 07:15:37 +0100 |
| commit | b8a10425ad550b04ccf3b5ff2493714615d7df4b (patch) | |
| tree | eab9ad609e34e4bad23cedc81ee338fe00961c3f /candle-examples/examples/custom-ops/cuda_kernels.rs | |
| parent | 5f20acf0804a624d6c274e488c897fb88d698f1a (diff) | |
| download | candle-b8a10425ad550b04ccf3b5ff2493714615d7df4b.tar.gz candle-b8a10425ad550b04ccf3b5ff2493714615d7df4b.tar.bz2 candle-b8a10425ad550b04ccf3b5ff2493714615d7df4b.zip | |
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.
Diffstat (limited to 'candle-examples/examples/custom-ops/cuda_kernels.rs')
| -rw-r--r-- | candle-examples/examples/custom-ops/cuda_kernels.rs | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/candle-examples/examples/custom-ops/cuda_kernels.rs b/candle-examples/examples/custom-ops/cuda_kernels.rs new file mode 100644 index 00000000..07d18342 --- /dev/null +++ b/candle-examples/examples/custom-ops/cuda_kernels.rs @@ -0,0 +1 @@ +pub const LAYERNORM_KERNELS: &str = include_str!(concat!(env!("OUT_DIR"), "/examples/custom-ops/kernels//layernorm_kernels.ptx")); |