Multiple updates and refactorings (#231)

2025-11-21 17:49:47 +08:00
parent bb4424aad4
commit 38f8ef73a4
80 changed files with 3767 additions and 2103 deletions
@@ -0,0 +1,10 @@
+#pragma once
+
+#include <torch/version.h>
+#include <cuda.h>
+
+// `torch::kFloat8_e4m3fn` is supported since PyTorch 2.1
+#define DG_FP8_COMPATIBLE (TORCH_VERSION_MAJOR > 2 or (TORCH_VERSION_MAJOR == 2 and TORCH_VERSION_MINOR >= 1))
+
+// `cuTensorMapEncodeTiled` is supported since CUDA Driver API 12.1
+#define DG_TENSORMAP_COMPATIBLE (CUDA_VERSION >= 12010)