Home
last modified time | relevance | path

Searched refs:CUfunction_attribute (Results 1 – 10 of 10) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/cuda/detail/
H A DLazyNVRTC.cpp170 CUDA_STUB3(cuFuncSetAttribute, CUfunction, CUfunction_attribute, int);
171 CUDA_STUB3(cuFuncGetAttribute, int*, CUfunction_attribute, CUfunction);
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/gpu/
H A Dgpu_types.h70 using GpuFunctionAttribute = CUfunction_attribute;
/aosp_15_r20/hardware/google/aemu/third-party/cuda/include/host-common/
Ddynlink_cuda_cuda.h265 } CUfunction_attribute; typedef
1413 typedef CUresult CUDAAPI tcuFuncGetAttribute(int *pi, CUfunction_attribute attrib, CUfunction hfunc…
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/cuda/
H A Dcuda_9_0.inc1130 CUresult CUDAAPI cuFuncGetAttribute(int *pi, CUfunction_attribute attrib,
1132 using FuncPtr = CUresult(CUDAAPI *)(int *, CUfunction_attribute, CUfunction);
1139 CUfunction_attribute attrib, int value) {
1140 using FuncPtr = CUresult(CUDAAPI *)(CUfunction, CUfunction_attribute, int);
H A Dcuda_10_1.inc1268 CUresult CUDAAPI cuFuncGetAttribute(int *pi, CUfunction_attribute attrib,
1270 using FuncPtr = CUresult(CUDAAPI *)(int *, CUfunction_attribute, CUfunction);
1277 CUfunction_attribute attrib, int value) {
1278 using FuncPtr = CUresult(CUDAAPI *)(CUfunction, CUfunction_attribute, int);
H A Dcuda_10_0.inc1249 CUresult CUDAAPI cuFuncGetAttribute(int *pi, CUfunction_attribute attrib,
1251 using FuncPtr = CUresult(CUDAAPI *)(int *, CUfunction_attribute, CUfunction);
1258 CUfunction_attribute attrib, int value) {
1259 using FuncPtr = CUresult(CUDAAPI *)(CUfunction, CUfunction_attribute, int);
H A Dcuda_10_2.inc1389 CUresult CUDAAPI cuFuncGetAttribute(int *pi, CUfunction_attribute attrib,
1391 using FuncPtr = CUresult(CUDAAPI *)(int *, CUfunction_attribute, CUfunction);
1398 CUfunction_attribute attrib, int value) {
1399 using FuncPtr = CUresult(CUDAAPI *)(CUfunction, CUfunction_attribute, int);
H A Dcuda_11_2.inc1634 CUresult CUDAAPI cuFuncGetAttribute(int *pi, CUfunction_attribute attrib,
1636 using FuncPtr = CUresult(CUDAAPI *)(int *, CUfunction_attribute, CUfunction);
1643 CUfunction_attribute attrib, int value) {
1644 using FuncPtr = CUresult(CUDAAPI *)(CUfunction, CUfunction_attribute, int);
H A Dcuda_11_0.inc1676 CUresult CUDAAPI cuFuncGetAttribute(int *pi, CUfunction_attribute attrib,
1678 using FuncPtr = CUresult(CUDAAPI *)(int *, CUfunction_attribute, CUfunction);
1685 CUfunction_attribute attrib, int value) {
1686 using FuncPtr = CUresult(CUDAAPI *)(CUfunction, CUfunction_attribute, int);
H A Dcuda_driver.cc429 CUfunction_attribute attribute, CUfunction func, int* attribute_value) { in FuncGetAttribute()