From 9809955b483d61ce7c054897ada3bf117e4605c6 Mon Sep 17 00:00:00 2001 From: Yaxing Cai Date: Tue, 13 Jan 2026 01:18:30 +0800 Subject: [PATCH 1/2] Add unified_api.h include to device_guard.h --- include/tvm/ffi/extra/cuda/device_guard.h | 1 + 1 file changed, 1 insertion(+) diff --git a/include/tvm/ffi/extra/cuda/device_guard.h b/include/tvm/ffi/extra/cuda/device_guard.h index 01586886..e655e1ef 100644 --- a/include/tvm/ffi/extra/cuda/device_guard.h +++ b/include/tvm/ffi/extra/cuda/device_guard.h @@ -24,6 +24,7 @@ #define TVM_FFI_EXTRA_CUDA_DEVICE_GUARD_H_ #include +#include namespace tvm { namespace ffi { From f139412c631ca2aec750515c5599886ce4978771 Mon Sep 17 00:00:00 2001 From: Yaxing Cai Date: Tue, 13 Jan 2026 01:37:00 +0800 Subject: [PATCH 2/2] Update include/tvm/ffi/extra/cuda/device_guard.h apply gemini suggestion Co-authored-by: gemini-code-assist[bot] <176961590+gemini-code-assist[bot]@users.noreply.github.com> --- include/tvm/ffi/extra/cuda/device_guard.h | 1 - 1 file changed, 1 deletion(-) diff --git a/include/tvm/ffi/extra/cuda/device_guard.h b/include/tvm/ffi/extra/cuda/device_guard.h index e655e1ef..1b44cf6f 100644 --- a/include/tvm/ffi/extra/cuda/device_guard.h +++ b/include/tvm/ffi/extra/cuda/device_guard.h @@ -23,7 +23,6 @@ #ifndef TVM_FFI_EXTRA_CUDA_DEVICE_GUARD_H_ #define TVM_FFI_EXTRA_CUDA_DEVICE_GUARD_H_ -#include #include namespace tvm {