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(u8, bool, f16, f32);
impl_float_out_binary_promote(u8, i8, f16, f32);
impl_float_out_binary_promote(u8, i16, f16, f32);
impl_float_out_binary_promote(u8, i32, f32, f32);
impl_float_out_binary_promote(u8, i64, f64, f64);
impl_float_out_binary_promote(u8, u8, f16, f32);
impl_float_out_binary_promote(u8, u16, f16, f32);
impl_float_out_binary_promote(u8, u32, f32, f32);
impl_float_out_binary_promote(u8, u64, f64, f64);
impl_float_out_binary_promote(u8, f16, f16, f32);
impl_float_out_binary_promote(u8, bf16, bf16, f32);
impl_float_out_binary_promote(u8, f32, f32, f32);
impl_float_out_binary_promote(u8, f64, f64, f64);

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

impl_float_out_unary_promote(u8, f16, f32);