vllm/csrc/cuda_utils.h

#pragma once

#include <torch/extension.h>

int get_device_attribute(
    int attribute,
    int device_id);
Avoid multiple redefinition (#1817) 2023-12-15 01:35:58 +08:00			`#pragma once`

[Build] Avoid building too many extensions (#1624) 2023-11-24 08:31:19 +08:00			`#include <torch/extension.h>`

			`int get_device_attribute(`
			`int attribute,`
			`int device_id);`