2023-12-15 01:35:58 +08:00
|
|
|
#pragma once
|
|
|
|
|
|
2023-11-24 08:31:19 +08:00
|
|
|
#include <torch/extension.h>
|
|
|
|
|
|
2024-05-22 15:18:41 +08:00
|
|
|
int get_device_attribute(int attribute, int device_id);
|
2024-01-27 04:41:10 +08:00
|
|
|
|
2024-05-22 15:18:41 +08:00
|
|
|
int get_max_shared_memory_per_block_device_attribute(int device_id);
|
