Skip to content

GitLab

Commit 904a9254 authored by zhuyue's avatar zhuyue
Browse files

Issue/670 - fix: correct macro for mccub/hccub conditional compilation.

parent 62797984
Show 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 @ 904a9254
#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