We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
2 parents 6279798 + 904a925 commit cad2d45Copy full SHA for cad2d45
src/infiniop/ops/layer_norm/metax/layer_norm_metax.maca
@@ -1,6 +1,6 @@
1
#include "../../../devices/metax/metax_common.h"
2
#include "layer_norm_metax.h"
3
-#ifdef ENABLE_METAX_API
+#ifdef ENABLE_METAX_MC_API
4
#include <mccub/block/block_reduce.cuh>
5
#else
6
#include <hccub/block/block_reduce.cuh>
0 commit comments