diff options
Diffstat (limited to 'tests/autodiff/autopybind-basic.slang')
| -rw-r--r-- | tests/autodiff/autopybind-basic.slang | 25 |
1 files changed, 25 insertions, 0 deletions
diff --git a/tests/autodiff/autopybind-basic.slang b/tests/autodiff/autopybind-basic.slang new file mode 100644 index 000000000..5d4409474 --- /dev/null +++ b/tests/autodiff/autopybind-basic.slang @@ -0,0 +1,25 @@ +//TEST:SIMPLE(filecheck=CUDA): -target cuda -line-directive-mode none +//TEST:SIMPLE(filecheck=TORCH): -target torch -line-directive-mode none + +// CUDA: __global__ void __kernel__myKernel(TensorView inValues_[[#]], TensorView outValues_[[#]]) +[AutoPyBindCUDA] +[CudaKernel] +void myKernel(TensorView<float> inValues, TensorView<float> outValues) +{ + if (cudaThreadIdx().x > 0) + return; + outValues.store(cudaThreadIdx().x, sin(inValues.load(cudaThreadIdx().x))); +} + +// TORCH: {{^SLANG_PRELUDE_EXPORT$}} +// TORCH-NEXT: void __kernel__myKernel(TensorView {{[[:alnum:]_]+}}, TensorView {{[[:alnum:]_]+}}); + +// TORCH: {{^SLANG_PRELUDE_EXPORT$}} +// TORCH-NEXT: void myKernel(std::tuple<uint32_t, uint32_t, uint32_t> {{[[:alnum:]_]+}}, std::tuple<uint32_t, uint32_t, uint32_t> {{[[:alnum:]_]+}}, torch::Tensor {{[[:alnum:]_]+}}, torch::Tensor {{[[:alnum:]_]+}}) + +// TORCH: {{^SLANG_PRELUDE_EXPORT$}} +// TORCH-NEXT: std::tuple<std::tuple<const char*, const char*, const char*, const char*>, std::tuple<const char*, const char*>, const char*, const char*> __funcinfo__myKernel() + +// TORCH: m.def("myKernel", &myKernel, "myKernel"); + +// TORCH: m.def("__funcinfo__myKernel", &__funcinfo__myKernel, "__funcinfo__myKernel");
\ No newline at end of file |