mirror of
https://github.com/vllm-project/vllm.git
synced 2026-06-06 00:16:14 +00:00
Chris Leonard
07aeaf9d4d
Signed-off-by: Mikayla Gawarecki <mikaylagawarecki@gmail.com> Signed-off-by: Chris Leonard <chleonar@redhat.com> Co-authored-by: Mikayla Gawarecki <mikaylagawarecki@gmail.com> Co-authored-by: Shengqi Chen <harry-chen@outlook.com>
18 lines
548 B
C
18 lines
548 B
C
#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
|