sglang_v0.5.2/pytorch_2.8.0/torch/csrc/cuda/utils.h

10 lines
215 B
C++

#pragma once
#include <c10/cuda/CUDAStream.h>
#include <torch/csrc/utils/python_numbers.h>
#include <vector>
std::vector<std::optional<at::cuda::CUDAStream>>
THPUtils_PySequence_to_CUDAStreamList(PyObject* obj);