Is there a C++ and CUDA extension for jittor similar to include <torch/extention.h>?

Jittor / jittor

Jittor is a high-performance deep learning framework based on JIT compiling and meta-operators.

https://cg.cs.tsinghua.edu.cn/jittor/

Apache License 2.0

3.07k stars 308 forks source link

Is there a C++ and CUDA extension for jittor similar to include <torch/extention.h>? #497

Open xiazhi1 opened 9 months ago

xiazhi1 commented 9 months ago

Thanks for awesome work, I checked the official documentation of jittor and found that jittor does not seem to have CUDA/C++ extensions like torch/extension.h. Is my query method wrong? Can jittor implement the function of writing rasterizers interactively with CUDA code?