完整部署快照

upstream_ref/ds_vllm/csrc/torch_utils.h

@@ -0,0 +1,17 @@
+#pragma once
+
+// Shared TORCH_UTILS_CHECK across both libtorch stable and unstable source
+// files. Keep this header free of CUTLASS/CUTE so attention/quant headers can
+// use it.
+//
+// If TORCH_TARGET_VERSION is defined, we are building _C_stable_libtorch.so so
+// use STD_TORCH_CHECK via header-only.
+// Otherwise, use TORCH_CHECK via torch/all.h.
+
+#ifdef TORCH_TARGET_VERSION
+  #include <torch/headeronly/util/Exception.h>
+  #define TORCH_UTILS_CHECK STD_TORCH_CHECK
+#else
+  #include <torch/all.h>
+  #define TORCH_UTILS_CHECK TORCH_CHECK
+#endif