TorchInductor Playground

Investigate the compatibility of torchinductor with custom ops.

Build

Build a custom sigmoid

cd custom_cuda_op; python3 setup.py build; cd ..;

Verify the op

python3 verify_custom_cpp_ops.py

Verify various ways the op is integrated into existing torch.compile framework

python3 test_custom_op.py

Name		Name	Last commit message	Last commit date
Latest commit History 2 Commits
custom_cuda_op		custom_cuda_op
.gitignore		.gitignore
README.md		README.md
build.sh		build.sh
test_custom_op.py		test_custom_op.py
verify_custom_cpp_ops.py		verify_custom_cpp_ops.py