Home
last modified time | relevance | path

Searched refs:raw_descriptor (Results 1 – 15 of 15) sorted by relevance

/external/pytorch/aten/src/ATen/cuda/
DCUDASparseDescriptors.h84 cusparseMatDescr_t raw_descriptor = nullptr; in CuSparseMatDescriptor() local
85 TORCH_CUDASPARSE_CHECK(cusparseCreateMatDescr(&raw_descriptor)); in CuSparseMatDescriptor()
86 descriptor_.reset(raw_descriptor); in CuSparseMatDescriptor()
94 cusparseMatDescr_t raw_descriptor = nullptr; in CuSparseMatDescriptor() local
95 TORCH_CUDASPARSE_CHECK(cusparseCreateMatDescr(&raw_descriptor)); in CuSparseMatDescriptor()
96 TORCH_CUDASPARSE_CHECK(cusparseSetMatFillMode(raw_descriptor, fill_mode)); in CuSparseMatDescriptor()
97 TORCH_CUDASPARSE_CHECK(cusparseSetMatDiagType(raw_descriptor, diag_type)); in CuSparseMatDescriptor()
98 descriptor_.reset(raw_descriptor); in CuSparseMatDescriptor()
108 bsrsv2Info_t raw_descriptor = nullptr; in CuSparseBsrsv2Info() local
109 TORCH_CUDASPARSE_CHECK(cusparseCreateBsrsv2Info(&raw_descriptor)); in CuSparseBsrsv2Info()
[all …]
DCUDASparseDescriptors.cpp96 cusparseDnMatDescr_t raw_descriptor = nullptr; in createRawDnMatDescriptor() local
98 &raw_descriptor, in createRawDnMatDescriptor()
110 raw_descriptor, batch_count, input_strides[ndim - 3])); in createRawDnMatDescriptor()
112 return raw_descriptor; in createRawDnMatDescriptor()
135 cusparseDnVecDescr_t raw_descriptor = nullptr; in CuSparseDnVecDescriptor() local
137 &raw_descriptor, input.numel(), input.data_ptr(), value_type)); in CuSparseDnVecDescriptor()
138 descriptor_.reset(raw_descriptor); in CuSparseDnVecDescriptor()
172 cusparseSpMatDescr_t raw_descriptor = nullptr; in CuSparseSpMatCsrDescriptor() local
174 &raw_descriptor, // output descriptor in CuSparseSpMatCsrDescriptor()
203 raw_descriptor, in CuSparseSpMatCsrDescriptor()
[all …]
DCUDABlas.cpp273 cublasLtMatmulDesc_t raw_descriptor = nullptr; in CuBlasLtMatmulDescriptor() local
275 cublasLtMatmulDescCreate(&raw_descriptor, compute_type, scale_type)); in CuBlasLtMatmulDescriptor()
276 descriptor_.reset(raw_descriptor); in CuBlasLtMatmulDescriptor()
294 cublasLtMatrixLayout_t raw_descriptor = nullptr; in CuBlasLtMatrixLayout() local
296 cublasLtMatrixLayoutCreate(&raw_descriptor, type, t ? cols : rows, t ? rows : cols, ld)); in CuBlasLtMatrixLayout()
297 descriptor_.reset(raw_descriptor); in CuBlasLtMatrixLayout()
310 cublasLtMatmulPreference_t raw_descriptor = nullptr; in CuBlasLtMatmulPreference() local
311 TORCH_CUDABLAS_CHECK(cublasLtMatmulPreferenceCreate(&raw_descriptor)); in CuBlasLtMatmulPreference()
312 descriptor_.reset(raw_descriptor); in CuBlasLtMatmulPreference()
/external/pytorch/aten/src/ATen/mkl/
DSparseDescriptors.h99 sparse_matrix_t raw_descriptor; in MklSparseCsrDescriptor() local
110 &raw_descriptor, in MklSparseCsrDescriptor()
122 &raw_descriptor, in MklSparseCsrDescriptor()
132 descriptor_.reset(raw_descriptor); in MklSparseCsrDescriptor()
136 sparse_matrix_t raw_descriptor = nullptr; in MklSparseCsrDescriptor() local
137 descriptor_.reset(raw_descriptor); in MklSparseCsrDescriptor()
/external/avb/rust/src/descriptor/
Dmod.rs106 unsafe fn new(raw_descriptor: *const AvbDescriptor) -> DescriptorResult<Self> { in new()
113 if !unsafe { avb_descriptor_validate_and_byteswap(raw_descriptor, &mut descriptor) } { in new()
129 let contents = unsafe { slice::from_raw_parts(raw_descriptor as *const u8, total_size) }; in new()
/external/crosvm/gpu_display/src/gpu_display_win/
Dwindow_procedure_thread.rs109 let raw_descriptor = handle.as_raw_descriptor(); in add() localVariable
110 if self.triggers.contains_key(&raw_descriptor) { in add()
114 self.triggers.insert(raw_descriptor, token); in add()
115 self.raw_handles.push(raw_descriptor); in add()
/external/crosvm/base/src/sys/linux/
Dmod.rs468 pub fn validate_raw_descriptor(raw_descriptor: RawDescriptor) -> Result<RawDescriptor> { in validate_raw_descriptor()
469 validate_raw_fd(&raw_descriptor) in validate_raw_descriptor()
527 let raw_descriptor = path in safe_descriptor_from_path() localVariable
532 let validated_fd = validate_raw_fd(&raw_descriptor)?; in safe_descriptor_from_path()
/external/crosvm/rutabaga_gfx/src/
Dgfxstream.rs263 let raw_descriptor = stream_handle.os_handle as RawDescriptor; in export_fence() localVariable
267 let handle = unsafe { OwnedDescriptor::from_raw_descriptor(raw_descriptor) }; in export_fence()
505 let raw_descriptor = stream_handle.os_handle as RawDescriptor; in export_blob() localVariable
509 let handle = unsafe { OwnedDescriptor::from_raw_descriptor(raw_descriptor) }; in export_blob()
/external/pytorch/aten/src/ATen/cuda/tunable/
DGemmHipblaslt.h386 hipblasLtMatmulDesc_t raw_descriptor = nullptr; in HipBlasLtMatmulDescriptor() local
388 hipblasLtMatmulDescCreate(&raw_descriptor, compute_type, scale_type)); in HipBlasLtMatmulDescriptor()
389 descriptor_.reset(raw_descriptor); in HipBlasLtMatmulDescriptor()
/external/crosvm/rutabaga_gfx/kumquat/gpu_client/src/virtgpu/
Dvirtgpu_kumquat.rs496 raw_descriptor: &mut RutabagaRawDescriptor, in submit_command()
569 *raw_descriptor = fence_opt in submit_command()
/external/perfetto/src/trace_processor/perfetto_sql/intrinsics/operators/
Dspan_join_operator.h99 static base::Status Parse(const std::string& raw_descriptor,
Dspan_join_operator.cc539 const std::string& raw_descriptor, in Parse() argument
545 base::StringSplitter splitter(raw_descriptor, ' '); in Parse()
/external/crosvm/devices/src/virtio/fs/
Dpassthrough.rs390 let raw_descriptor = unsafe { in open_fscreate() localVariable
400 if raw_descriptor < 0 { in open_fscreate()
408 unsafe { File::from_raw_descriptor(raw_descriptor) } in open_fscreate()
766 let raw_descriptor = syscall!(unsafe { in new() localVariable
791 let proc = unsafe { File::from_raw_descriptor(raw_descriptor) }; in new()
919 let raw_descriptor = syscall!(unsafe { in open_fd() localVariable
928 Ok(unsafe { File::from_raw_descriptor(raw_descriptor) }) in open_fd()
2154 let raw_descriptor = unsafe { libc::openat64(libc::AT_FDCWD, root.as_ptr(), flags) }; in init() localVariable
2155 if raw_descriptor < 0 { in init()
2160 let f = unsafe { File::from_raw_descriptor(raw_descriptor) }; in init()
/external/crosvm/devices/src/
Dvfio.rs697 let raw_descriptor = unsafe { libc::dup(self.container.as_raw_descriptor()) }; in clone_as_raw_descriptor() localVariable
698 if raw_descriptor < 0 { in clone_as_raw_descriptor()
701 Ok(raw_descriptor) in clone_as_raw_descriptor()
/external/crosvm/devices/src/virtio/
Dwl.rs1990 if let Some(raw_descriptor) = mapper.as_raw_descriptor() { in keep_rds()
1991 keep_rds.push(raw_descriptor); in keep_rds()