Searched refs:workspace_ptr (Results 1 – 7 of 7) sorted by relevance
| /external/pytorch/torch/_inductor/codegen/rocm/ |
| D | rocm_benchmark_request.py | 67 workspace_ptr = c_void_p(0)
74 workspace_ptr = c_void_p(self.workspace.data_ptr())
82 workspace_ptr, # set workspace ptr,
|
| /external/pytorch/aten/src/ATen/native/cudnn/ |
| D | Conv_v8.cpp | 367 auto workspace_ptr = in run_conv_plan() local
396 .setWorkspacePointer(workspace_size ? workspace_ptr.get() : nullptr) in run_conv_plan()
414 auto workspace_ptr = in run_conv_plan_fused() local
421 .setWorkspacePointer(workspace_size ? workspace_ptr.get() : nullptr) in run_conv_plan_fused()
650 at::DataPtr& workspace_ptr) { in generate_and_filter_plans() argument
676 workspace_ptr = in generate_and_filter_plans()
720 at::DataPtr workspace_ptr; in get_plans_from_find() local
722 handle, opGraph, generator, x, valid_plans, workspace_ptr); in get_plans_from_find()
727 .setWorkspacePointer(workspace_ptr ? workspace_ptr.get() : nullptr) in get_plans_from_find()
775 at::DataPtr workspace_ptr; in get_plans_from_find_fused() local
[all …]
|
| D | MHA.cpp | 602 auto workspace_ptr = in run_cudnn_SDP_fprop() local
605 mha_graph->execute(handle, variant_pack, workspace_ptr.get()).is_good()); in run_cudnn_SDP_fprop()
695 auto workspace_ptr = in run_cudnn_SDP_bprop() local
697 TORCH_CHECK(!workspace_size || workspace_ptr.get()); in run_cudnn_SDP_bprop()
699 mha_graph->execute(handle, variant_pack, workspace_ptr.get()).is_good()); in run_cudnn_SDP_bprop()
|
| /external/pytorch/aten/src/ATen/native/quantized/cudnn/ |
| D | Linear.cpp | 147 auto workspace_ptr = c10::cuda::CUDACachingAllocator::get()->allocate(workspace_size); in apply_impl_helper() local
159 .setWorkspacePointer(workspace_size ? workspace_ptr.get() : nullptr) in apply_impl_helper()
|
| D | Conv.cpp | 119 auto workspace_ptr = c10::cuda::CUDACachingAllocator::get()->allocate(workspace_size); in apply_impl_helper() local
131 .setWorkspacePointer(workspace_size ? workspace_ptr.get() : nullptr) in apply_impl_helper()
|
| /external/pytorch/torch/_inductor/ |
| D | autotune_process.py | 729 workspace_ptr = c_void_p(0)
736 workspace_ptr = c_void_p(self.workspace.data_ptr())
744 workspace_ptr, # set workspace ptr,
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_dnn.cc | 4199 uint8* workspace_ptr = 0; in DoPoolBackward() local
4220 workspace_ptr = reinterpret_cast<uint8*>( in DoPoolBackward()
4268 workspace_ptr = reinterpret_cast<uint8*>(workspace.opaque()); in DoPoolBackward()
4276 output_diff_data.opaque(), workspace_ptr); in DoPoolBackward()
|