mirror of
https://github.com/NVIDIA/TensorRT-LLM.git
synced 2026-01-14 06:27:45 +08:00
32 lines
924 B
Diff
32 lines
924 B
Diff
From 2a440348958b3f0a2b09458bd76fe5959b371c0c Mon Sep 17 00:00:00 2001
|
|
From: eqy <eddiey@nvidia.com>
|
|
Date: Tue, 19 Dec 2023 05:56:48 +0000
|
|
Subject: [PATCH] [CUDA] Include `<thrust/swap.h>` in `LinearAlgebra.cu`
|
|
(#116072)
|
|
|
|
Fixes build against the latest `NVIDIA/cccl`.
|
|
|
|
CC @malfet @xwang233 @ptrblck
|
|
|
|
Pull Request resolved: https://github.com/pytorch/pytorch/pull/116072
|
|
Approved by: https://github.com/malfet, https://github.com/xwang233
|
|
---
|
|
aten/src/ATen/native/cuda/LinearAlgebra.cu | 2 ++
|
|
1 file changed, 2 insertions(+)
|
|
|
|
diff --git a/aten/src/ATen/native/cuda/LinearAlgebra.cu b/aten/src/ATen/native/cuda/LinearAlgebra.cu
|
|
index fb59f976041..a6a566a5de2 100644
|
|
--- a/aten/src/ATen/native/cuda/LinearAlgebra.cu
|
|
+++ b/aten/src/ATen/native/cuda/LinearAlgebra.cu
|
|
@@ -9,6 +9,8 @@
|
|
#include <ATen/native/ReduceOps.h>
|
|
#include <c10/core/Scalar.h>
|
|
|
|
+#include <thrust/swap.h>
|
|
+
|
|
namespace at::native {
|
|
|
|
namespace {
|
|
--
|
|
2.34.1
|