Searched defs:is_half (Results 1 – 4 of 4) sorted by relevance
455 const auto is_half = input.second.has_value() && in generateKernel() local589 const auto is_half = (output.second.scalar_type == at::ScalarType::Half); in generateKernel() local
57 bool use_tensor_cores(sdp_params const& params, cudaDeviceProp* dprops, bool is_half) { in use_tensor_cores()68 bool is_half = (params.query.dtype() == at::kHalf) || in minimum_gemm_alignment() local
195 bool is_half = !cutlass::platform::is_same<scalar_t, float>::value; in getWarpsPerSmBw() local
1128 is_half(struct ir3_instruction *instr) in is_half() function