I try to use ggml matmul together with pytorch. But I fail to use pytorch cudagraph to capture the matmul

[Eutenacity]

I try to use ggml matmul together with pytorch. I put my code in the https://github.com/Eutenacity/python-ggml-matmul
I hope this can help.
But I can not use PyTorch cudagraph to capture the matmul.
If anyone can give me a favor