Skip to content

GitLab

Unverified Commit cad2d45a authored by PanZezhong1725's avatar PanZezhong1725 Committed by GitHub
Browse files

Merge pull request #671 from gongchensu/feature/metax_op_fixed 

fix: correct macro for mccub/hccub conditional compilation.
parents 62797984 904a9254
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
src/infiniop/ops/layer_norm/metax/layer_norm_metax.maca
View file @ cad2d45a
#include "../../../devices/metax/metax_common.h"
#include "layer_norm_metax.h"
#ifdef ENABLE_METAX_API
#ifdef ENABLE_METAX_MC_API
#include <mccub/block/block_reduce.cuh>
#else
#include <hccub/block/block_reduce.cuh>
......
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