File tree Expand file tree Collapse file tree 2 files changed +11
-11
lines changed
kernels/portable/cpu/util Expand file tree Collapse file tree 2 files changed +11
-11
lines changed Original file line number Diff line number Diff line change @@ -662,16 +662,16 @@ Error defineConvertNode(
662662 MAYBE_UNUSED (flatbuffer_graph);
663663 auto graph_node = node->xnode_union_as_XNNConvert ();
664664#ifdef ENABLE_XNNPACK_KLEIDI
665- // This is not currently exposed at include/xnnpack.h yet once it is
666- // we can remove this runtime logic and do this ahead-of-time
667- #define XNN_FLAG_MAYBE_PACK_FOR_QB4W_GEMM 0x00000100 ;
668- if (isQP8 (flatbuffer_graph, node)) {
669- flags |= XNN_FLAG_MAYBE_PACK_FOR_QB4W_GEMM;
670- ET_LOG (
671- Debug,
672- " Setting XNN_FLAG_MAYBE_PACK_FOR_QB4W_GEMM flag for convert node %i" ,
673- node->debug_handle ());
674- }
665+ // This is not currently exposed at include/xnnpack.h yet once it is
666+ // we can remove this runtime logic and do this ahead-of-time
667+ #define XNN_FLAG_MAYBE_PACK_FOR_QB4W_GEMM 0x00000100 ;
668+ if (isQP8 (flatbuffer_graph, node)) {
669+ flags |= XNN_FLAG_MAYBE_PACK_FOR_QB4W_GEMM;
670+ ET_LOG (
671+ Debug,
672+ " Setting XNN_FLAG_MAYBE_PACK_FOR_QB4W_GEMM flag for convert node %i" ,
673+ node->debug_handle ());
674+ }
675675#endif
676676
677677 int32_t flags = graph_node->flags ();
Original file line number Diff line number Diff line change @@ -832,7 +832,7 @@ template <typename Func>
832832 std::optional<ArrayRef<int64_t >> dim_list,
833833 const Tensor& out,
834834 const Func& func) {
835- #ifdef ET_UE_THREADPOOL
835+ #ifdef ET_USE_THREADPOOL
836836 const ssize_t reduction_size = get_reduced_dim_product (in, dim_list);
837837 const auto grain_size = std::max (
838838 static_cast <ssize_t >(1 ),
You can’t perform that action at this time.
0 commit comments