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
36

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


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

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

impl_float_out_unary_promote(i16, f16, f32);