hpt-cudakernels 0.1.3

A library implements cuda kernels for hpt
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35

#pragma once
#include "cuda_bf16.h"
#include "cuda_fp16.h"
#include "../macros.cuh"
#include "stdint.h"

impl_float_out_binary_promote(bf16, bool, bf16, f32);
impl_float_out_binary_promote(bf16, i8, bf16, f32);
impl_float_out_binary_promote(bf16, i16, bf16, f32);
impl_float_out_binary_promote(bf16, i32, f32, f32);
impl_float_out_binary_promote(bf16, i64, f64, f64);
impl_float_out_binary_promote(bf16, u8, bf16, f32);
impl_float_out_binary_promote(bf16, u16, bf16, f32);
impl_float_out_binary_promote(bf16, u32, f32, f32);
impl_float_out_binary_promote(bf16, u64, f64, f64);
impl_float_out_binary_promote(bf16, bf16, bf16, f32);
impl_float_out_binary_promote(bf16, f16, bf16, f32);
impl_float_out_binary_promote(bf16, f32, f32, f32);
impl_float_out_binary_promote(bf16, f64, f64, f64);

impl_normal_out_promote(bf16, bool, bf16, f32);
impl_normal_out_promote(bf16, i8, bf16, f32);
impl_normal_out_promote(bf16, i16, bf16, f32);
impl_normal_out_promote(bf16, i32, f32, f32);
impl_normal_out_promote(bf16, i64, f64, f64);
impl_normal_out_promote(bf16, u8, bf16, f32);
impl_normal_out_promote(bf16, u16, bf16, f32);
impl_normal_out_promote(bf16, u32, f32, f32);
impl_normal_out_promote(bf16, u64, f64, f64);
impl_normal_out_promote(bf16, bf16, bf16, f32);
impl_normal_out_promote(bf16, f16, bf16, f32);
impl_normal_out_promote(bf16, f32, f32, f32);
impl_normal_out_promote(bf16, f64, f64, f64);

impl_float_out_unary_promote(bf16, bf16, f32);