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

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

impl_float_out_unary_promote(u32, f32, f32);