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-14 12:35:58 -05:00			`#pragma once`

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

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