Searched defs:dprops (Results 1 – 7 of 7) sorted by relevance
57 bool use_tensor_cores(sdp_params const& params, cudaDeviceProp* dprops, bool is_half) { in use_tensor_cores()67 auto dprops = at::cuda::getCurrentDeviceProperties(); in minimum_gemm_alignment() local173 bool check_sm_version(cudaDeviceProp * dprops) { in check_sm_version()190 auto dprops = at::cuda::getCurrentDeviceProperties(); in check_flash_attention_hardware_support() local198 auto dprops = at::cuda::getCurrentDeviceProperties(); in check_flash_attention_hardware_support() local221 auto dprops = at::cuda::getCurrentDeviceProperties(); in check_mem_efficient_hardware_support() local229 auto dprops = at::cuda::getCurrentDeviceProperties(); in check_mem_efficient_hardware_support() local252 auto dprops = at::cuda::getCurrentDeviceProperties(); in check_requires_grad_and_head_dim_gt192_constraints_on_sm86_89() local439 auto dprops = at::cuda::getCurrentDeviceProperties(); in check_cudnn_hardware_support() local
209 auto dprops = at::cuda::getCurrentDeviceProperties(); in run_mha_fwd_hdim96() local235 auto dprops = at::cuda::getCurrentDeviceProperties(); in run_mha_fwd_hdim128() local272 auto dprops = at::cuda::getCurrentDeviceProperties(); in run_mha_fwd_hdim160() local
300 const int num_splits, cudaDeviceProp *dprops, struct c10::TensorOptions opts) { in set_params_splitkv()363 auto dprops = at::cuda::getCurrentDeviceProperties(); in mha_fwd() local563 auto dprops = at::cuda::getCurrentDeviceProperties(); in mha_varlen_fwd() local830 auto dprops = at::cuda::getCurrentDeviceProperties(); in mha_bwd() local1048 auto dprops = at::cuda::getCurrentDeviceProperties(); in mha_varlen_bwd() local1273 auto dprops = at::cuda::getCurrentDeviceProperties(); in mha_fwd_kvcache() local
80 auto dprops = at::cuda::getCurrentDeviceProperties(); in run_flash_bwd_seqk_parallel() local
636 const auto dprops = at::cuda::getCurrentDeviceProperties(); in _sparse_semi_structured_linear() local
373 auto dprops = at::cuda::getCurrentDeviceProperties(); in bmm_nested_cuda() local
863 auto dprops = at::cuda::getCurrentDeviceProperties(); in _scaled_mm_allowed_device() local