Function round_to_bfloat16

src/framework/sampling/torch_random.cpp:146–155

    grid = std::max<uint64_t>(1, std::min(grid_cap, grid));
    return block_size * grid;
}

float round_to_bfloat16(float value) {
    uint32_t bits = 0;
    static_assert(sizeof(bits) == sizeof(value));
    std::memcpy(&bits, &value, sizeof(bits));
    bits += 0x7FFFU + ((bits >> 16U) & 1U);
    bits &= 0xFFFF0000U;
    float rounded = 0.0F;
    std::memcpy(&rounded, &bits, sizeof(rounded));
    return rounded;
}