156 lines
9.9 KiB
C++
156 lines
9.9 KiB
C++
#pragma once
|
|
|
|
#include <c10/xpu/XPUMacros.h>
|
|
#include <sycl/sycl.hpp>
|
|
|
|
namespace c10::xpu {
|
|
|
|
#define AT_FORALL_XPU_DEVICE_PROPERTIES(_) \
|
|
/* the device name of this SYCL device. */ \
|
|
_(name) \
|
|
\
|
|
/* the device type associated with the device. */ \
|
|
_(device_type) \
|
|
\
|
|
/* the vendor of this SYCL device. */ \
|
|
_(vendor) \
|
|
\
|
|
/* a backend-defined driver version as a std::string. */ \
|
|
_(driver_version) \
|
|
\
|
|
/* the SYCL version as a std::string in the form <major>.<minor> */ \
|
|
_(version) \
|
|
\
|
|
/* true if the SYCL device is available. Otherwise, return false. */ \
|
|
_(is_available) \
|
|
\
|
|
/* the maximum size in bytes of the arguments that can be passed to a \
|
|
* kernel. */ \
|
|
_(max_parameter_size) \
|
|
\
|
|
/* the number of parallel compute units available to the device. */ \
|
|
_(max_compute_units) \
|
|
\
|
|
/* the maximum dimensions that specify the global and local work-item IDs \
|
|
* used by the data parallel execution model. */ \
|
|
_(max_work_item_dimensions) \
|
|
\
|
|
/* the maximum number of workitems that are permitted in a work-group \
|
|
* executing a kernel on a single compute unit. */ \
|
|
_(max_work_group_size) \
|
|
\
|
|
/* the maximum number of subgroups in a work-group for any kernel executed \
|
|
* on the device. */ \
|
|
_(max_num_sub_groups) \
|
|
\
|
|
/* a std::vector of size_t containing the set of sub-group sizes supported \
|
|
* by the device. */ \
|
|
_(sub_group_sizes) \
|
|
\
|
|
/* the maximum configured clock frequency of this SYCL device in MHz. */ \
|
|
_(max_clock_frequency) \
|
|
\
|
|
/* the default compute device address space size specified as an unsigned \
|
|
* integer value in bits. Must return either 32 or 64. */ \
|
|
_(address_bits) \
|
|
\
|
|
/* the maximum size of memory object allocation in bytes. */ \
|
|
_(max_mem_alloc_size) \
|
|
\
|
|
/* the minimum value in bits of the largest supported SYCL built-in data \
|
|
* type if this SYCL device is not of device type \
|
|
* sycl::info::device_type::custom. */ \
|
|
_(mem_base_addr_align) \
|
|
\
|
|
/* a std::vector of info::fp_config describing the half/single/double \
|
|
* precision floating-point capability of this SYCL device. */ \
|
|
_(half_fp_config) \
|
|
_(single_fp_config) \
|
|
_(double_fp_config) \
|
|
\
|
|
/* the size of global device memory in bytes. */ \
|
|
_(global_mem_size) \
|
|
\
|
|
/* the type of global memory cache supported. */ \
|
|
_(global_mem_cache_type) \
|
|
\
|
|
/* the size of global memory cache in bytes. */ \
|
|
_(global_mem_cache_size) \
|
|
\
|
|
/* the size of global memory cache line in bytes. */ \
|
|
_(global_mem_cache_line_size) \
|
|
\
|
|
/* the type of local memory supported. */ \
|
|
_(local_mem_type) \
|
|
\
|
|
/* the size of local memory arena in bytes. */ \
|
|
_(local_mem_size) \
|
|
\
|
|
/* the maximum number of sub-devices that can be created when this device is \
|
|
* partitioned. */ \
|
|
_(partition_max_sub_devices) \
|
|
\
|
|
/* the resolution of device timer in nanoseconds. */ \
|
|
_(profiling_timer_resolution) \
|
|
\
|
|
/* the preferred native vector width size for built-in scalar types that can \
|
|
* be put into vectors. */ \
|
|
_(preferred_vector_width_char) \
|
|
_(preferred_vector_width_short) \
|
|
_(preferred_vector_width_int) \
|
|
_(preferred_vector_width_long) \
|
|
_(preferred_vector_width_float) \
|
|
_(preferred_vector_width_double) \
|
|
_(preferred_vector_width_half) \
|
|
\
|
|
/* the native ISA vector width. The vector width is defined as the number of \
|
|
* scalar elements that can be stored in the vector. */ \
|
|
_(native_vector_width_char) \
|
|
_(native_vector_width_short) \
|
|
_(native_vector_width_int) \
|
|
_(native_vector_width_long) \
|
|
_(native_vector_width_float) \
|
|
_(native_vector_width_double) \
|
|
_(native_vector_width_half)
|
|
|
|
#define AT_FORALL_XPU_EXT_DEVICE_PROPERTIES(_) \
|
|
/* the number of EUs associated with the Intel GPU. */ \
|
|
_(gpu_eu_count, 512) \
|
|
\
|
|
/* the number of EUs in a subslice. */ \
|
|
_(gpu_eu_count_per_subslice, 8) \
|
|
\
|
|
/* the simd width of EU of GPU. */ \
|
|
_(gpu_eu_simd_width, 8) \
|
|
\
|
|
/* the number of hardware threads per EU of GPU. */ \
|
|
_(gpu_hw_threads_per_eu, 8)
|
|
|
|
#define _DEFINE_SYCL_PROP(ns, property, member) \
|
|
ns::property::return_type member;
|
|
|
|
#define DEFINE_DEVICE_PROP(property) \
|
|
_DEFINE_SYCL_PROP(sycl::info::device, property, property)
|
|
|
|
#define DEFINE_PLATFORM_PROP(property, member) \
|
|
_DEFINE_SYCL_PROP(sycl::info::platform, property, member)
|
|
|
|
#define DEFINE_EXT_DEVICE_PROP(property, ...) \
|
|
_DEFINE_SYCL_PROP(sycl::ext::intel::info::device, property, property)
|
|
|
|
struct C10_XPU_API DeviceProp {
|
|
AT_FORALL_XPU_DEVICE_PROPERTIES(DEFINE_DEVICE_PROP);
|
|
|
|
// the platform name.
|
|
DEFINE_PLATFORM_PROP(name, platform_name);
|
|
|
|
AT_FORALL_XPU_EXT_DEVICE_PROPERTIES(DEFINE_EXT_DEVICE_PROP)
|
|
};
|
|
|
|
#undef _DEFINE_SYCL_PROP
|
|
#undef DEFINE_DEVICE_PROP
|
|
#undef DEFINE_PLATFORM_PROP
|
|
#undef DEFINE_EXT_DEVICE_PROP
|
|
|
|
} // namespace c10::xpu
|