Unverified Commit 685b6e03 authored by crapromer's avatar crapromer Committed by GitHub
Browse files

Issue/636 - Fix metax compile error of fp8 with maca (#639)

* issue/636 - add support for fp8 with maca sdk

* issue/636 - add functional header to support Fn

* issue/636 - format code with clang
parent 9097bb84
......@@ -10,6 +10,7 @@
#include <hcblas/hcblas.h>
#include <hcdnn/hcdnn.h>
#endif
#include <functional>
#include <memory>
#define CHECK_MCBLAS(API) CHECK_INTERNAL(API, HCBLAS_STATUS_SUCCESS)
......
......@@ -5,6 +5,7 @@
#define hcdnnDataType_t mcdnnDataType_t
#define hpcc_bfloat16 maca_bfloat16
#define hpcc_bfloat162 maca_bfloat162
#define __hpcc_fp8_e4m3 __maca_fp8_e4m3
#define __hpcc_bfloat16 __maca_bfloat16
#define __hpcc_bfloat16 __maca_bfloat16
#define hcError_t mcError_t
......
#define INFINIOP_METAX_KERNEL __global__ void
#ifdef ENABLE_METAX_MC_API
#include <maca_fp8.h>
#else
#include <hpcc_fp8.h>
#endif
// Posible maximum number of threads per block for METAX architectures
// Used for picking correct kernel launch configuration
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment