Skip to content

GitLab

"git@developer.sourcefind.cn:tsoc/superbenchmark.git" did not exist on "37a70cbe29a9f852cca08ad758332420297d95f4"
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 "../../../devices/metax/metax_common.h"
#include "layer_norm_metax.h" #include "layer_norm_metax.h"
#ifdef ENABLE_METAX_API #ifdef ENABLE_METAX_MC_API
#include <mccub/block/block_reduce.cuh> #include <mccub/block/block_reduce.cuh>
#else #else
#include <hccub/block/block_reduce.cuh> #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