feat: support libtorch

2025-10-21 09:55:33 +00:00
parent 239112cb4c
commit 301cbc1d75
22 changed files with 652 additions and 245 deletions
@@ -1,7 +1,6 @@
 #pragma once

 #include <cute/arch/mma_sm100_umma.hpp>
-#include <torch/python.h>

 #include "math.hpp"
 #include "exception.hpp"
@@ -1,6 +1,6 @@
 #pragma once

-#include <torch/python.h>
+#include <cstdint>

 #include "exception.hpp"