[ROCM] enable matmul(dot) and others (#391)

This commit is contained in:
Michael Melesse
2021-12-13 12:28:15 -08:00
committed by GitHub
parent 73b04d71b2
commit 94d5c2e8b5
12 changed files with 251 additions and 52 deletions

View File

@@ -11,7 +11,6 @@
#include "triton/external/CUDA/nvml.h"
//// HIP backend
//#define __HIP_PLATFORM_AMD__
#include "triton/external/hip.h"
//Exceptions

View File

@@ -42,6 +42,7 @@ public:
value *get_int64(int64_t val);
value *get_float16(float val);
value *get_float32(float val);
value *get_float64(float val);
value *get_range(int32_t lo, int32_t hi);
// Types
type *get_void_ty();