Searched defs:ncclDataType (Results 1 – 2 of 2) sorted by relevance
60 std::map<at::ScalarType, ncclDataType_t> ncclDataType = { variable3356 auto ncclDataType = getNcclDataType(input.scalar_type()); in checkForNCCLErrorsInternal() local3420 auto ncclDataType = getNcclDataType(input.scalar_type()); in checkForNCCLErrorsInternal() local3519 auto ncclDataType = getNcclDataType(input.scalar_type()); in checkForNCCLErrorsInternal() local3671 auto ncclDataType = getNcclDataType(input.scalar_type()); in checkForNCCLErrorsInternal() local3712 const auto ncclDataType = getNcclDataType(input.scalar_type()); in checkForNCCLErrorsInternal() local3904 const auto ncclDataType = getNcclDataType(input.scalar_type()); in checkForNCCLErrorsInternal() local4023 auto ncclDataType = getNcclDataType(input.scalar_type()); in checkForNCCLErrorsInternal() local4058 auto ncclDataType = getNcclDataType(input.scalar_type()); in checkForNCCLErrorsInternal() local
56 enum class ncclDataType { enum