Skip to content

Commit

Permalink
Merge branch 'master' into hunyuan_model
Browse files Browse the repository at this point in the history
  • Loading branch information
zhaohb authored Dec 27, 2024
2 parents 9e3ba96 + c14134a commit fc4a121
Show file tree
Hide file tree
Showing 43 changed files with 112 additions and 0 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -26,6 +26,18 @@
} \
}

#define FUNC_LOAD_LEFTOVERS(inner, outer) unroll_for (uint lh = 0; lh < outer; ++lh) { \
const uint input_idx = INPUT0_GET_TILED_INDEX(INPUT0_TILED_ORDER); \
INPUTVTYPE read_data; \
unroll_for (uint lw = 0; lw < inner; ++lw) { \
read_data[lw] = input[input_idx + lw]; \
} \
unroll_for (uint lw = 0; lw < inner; ++lw) { \
const uint dst = local_buf_offset + lw; \
transpose_buf[dst][lh] = read_data[lw]; \
} \
}

#define FUNC_VSTORE(loop) unroll_for (uint lw = 0; lw < loop; ++lw) { \
const uint output_idx = output_idx_tile + (lw * x_pitch); \
VSTORE(TO_OUTPUTVTYPE(transpose_buf[local_buf_offset + lw]), 0, output + output_idx); \
Expand Down Expand Up @@ -109,7 +121,15 @@ KERNEL (reorder_data_bfyx_to_blocked_format)(

if (F_NO_REMAINDER_CONDITION) {
// read and transpose
#ifdef X_REMAINDER_CONDITION
if (X_NO_REMAINDER_CONDITION) {
FUNC_VLOAD(TILE_SIZE, TILE_SIZE)
} else {
FUNC_LOAD_LEFTOVERS(X_REMAINDER_SIZE, TILE_SIZE)
}
#else
FUNC_VLOAD(TILE_SIZE, TILE_SIZE)
#endif

// write to ddr
#ifdef X_REMAINDER_CONDITION
Expand All @@ -125,7 +145,15 @@ KERNEL (reorder_data_bfyx_to_blocked_format)(
#ifdef F_REMAINDER_CONDITION
else if (F_REMAINDER_CONDITION) {
// read and transpose
#ifdef X_REMAINDER_CONDITION
if (X_NO_REMAINDER_CONDITION) {
FUNC_VLOAD(TILE_SIZE, F_REMAINDER_SIZE)
} else {
FUNC_LOAD_LEFTOVERS(X_REMAINDER_SIZE, F_REMAINDER_SIZE)
}
#else
FUNC_VLOAD(TILE_SIZE, F_REMAINDER_SIZE)
#endif

// write to ddr
#ifdef X_REMAINDER_CONDITION
Expand Down
2 changes: 2 additions & 0 deletions tests/layer_tests/onnx_tests/test_abs.py
Original file line number Diff line number Diff line change
Expand Up @@ -2,6 +2,8 @@
# SPDX-License-Identifier: Apache-2.0

import pytest
pytest.importorskip("openvino.tools.mo", reason="Ticket - 157136")

from common.layer_test_class import check_ir_version
from common.onnx_layer_test_class import OnnxRuntimeLayerTest, onnx_make_model

Expand Down
2 changes: 2 additions & 0 deletions tests/layer_tests/onnx_tests/test_and.py
Original file line number Diff line number Diff line change
Expand Up @@ -3,6 +3,8 @@

import numpy as np
import pytest
pytest.importorskip("openvino.tools.mo", reason="Ticket - 157136")

from common.layer_test_class import check_ir_version
from common.onnx_layer_test_class import OnnxRuntimeLayerTest, onnx_make_model

Expand Down
2 changes: 2 additions & 0 deletions tests/layer_tests/onnx_tests/test_argmax.py
Original file line number Diff line number Diff line change
Expand Up @@ -3,6 +3,8 @@

import numpy as np
import pytest
pytest.importorskip("openvino.tools.mo", reason="Ticket - 157136")

from common.layer_test_class import check_ir_version
from common.onnx_layer_test_class import OnnxRuntimeLayerTest, onnx_make_model

Expand Down
2 changes: 2 additions & 0 deletions tests/layer_tests/onnx_tests/test_ceil.py
Original file line number Diff line number Diff line change
Expand Up @@ -2,6 +2,8 @@
# SPDX-License-Identifier: Apache-2.0

import pytest
pytest.importorskip("openvino.tools.mo", reason="Ticket - 157136")

from common.layer_test_class import check_ir_version
from common.onnx_layer_test_class import OnnxRuntimeLayerTest, onnx_make_model

Expand Down
2 changes: 2 additions & 0 deletions tests/layer_tests/onnx_tests/test_clip.py
Original file line number Diff line number Diff line change
Expand Up @@ -2,6 +2,8 @@
# SPDX-License-Identifier: Apache-2.0

import pytest
pytest.importorskip("openvino.tools.mo", reason="Ticket - 157136")

from common.layer_test_class import check_ir_version
from common.onnx_layer_test_class import OnnxRuntimeLayerTest, onnx_make_model

Expand Down
2 changes: 2 additions & 0 deletions tests/layer_tests/onnx_tests/test_concat.py
Original file line number Diff line number Diff line change
Expand Up @@ -2,6 +2,8 @@
# SPDX-License-Identifier: Apache-2.0

import pytest
pytest.importorskip("openvino.tools.mo", reason="Ticket - 157136")

from common.layer_test_class import check_ir_version
from common.onnx_layer_test_class import OnnxRuntimeLayerTest, onnx_make_model

Expand Down
2 changes: 2 additions & 0 deletions tests/layer_tests/onnx_tests/test_conv.py
Original file line number Diff line number Diff line change
Expand Up @@ -3,6 +3,8 @@

import numpy as np
import pytest
pytest.importorskip("openvino.tools.mo", reason="Ticket - 157136")

from common.layer_test_class import check_ir_version
from common.onnx_layer_test_class import OnnxRuntimeLayerTest, onnx_make_model

Expand Down
2 changes: 2 additions & 0 deletions tests/layer_tests/onnx_tests/test_cumsum.py
Original file line number Diff line number Diff line change
Expand Up @@ -3,6 +3,8 @@

import numpy as np
import pytest
pytest.importorskip("openvino.tools.mo", reason="Ticket - 157136")

from common.layer_test_class import check_ir_version
from common.onnx_layer_test_class import OnnxRuntimeLayerTest, onnx_make_model

Expand Down
2 changes: 2 additions & 0 deletions tests/layer_tests/onnx_tests/test_dequantize_linear.py
Original file line number Diff line number Diff line change
Expand Up @@ -3,6 +3,8 @@

import numpy as np
import pytest
pytest.importorskip("openvino.tools.mo", reason="Ticket - 157136")

from common.layer_test_class import check_ir_version
from common.onnx_layer_test_class import OnnxRuntimeLayerTest, onnx_make_model

Expand Down
2 changes: 2 additions & 0 deletions tests/layer_tests/onnx_tests/test_elu.py
Original file line number Diff line number Diff line change
Expand Up @@ -2,6 +2,8 @@
# SPDX-License-Identifier: Apache-2.0

import pytest
pytest.importorskip("openvino.tools.mo", reason="Ticket - 157136")

from common.layer_test_class import check_ir_version
from common.onnx_layer_test_class import OnnxRuntimeLayerTest, onnx_make_model

Expand Down
2 changes: 2 additions & 0 deletions tests/layer_tests/onnx_tests/test_embedding_bag.py
Original file line number Diff line number Diff line change
Expand Up @@ -5,6 +5,8 @@

import numpy as np
import pytest
pytest.importorskip("openvino.tools.mo", reason="Ticket - 157136")

import torch
import torch.nn as nn
from common.layer_test_class import CommonLayerTest, check_ir_version
Expand Down
2 changes: 2 additions & 0 deletions tests/layer_tests/onnx_tests/test_floor.py
Original file line number Diff line number Diff line change
Expand Up @@ -2,6 +2,8 @@
# SPDX-License-Identifier: Apache-2.0

import pytest
pytest.importorskip("openvino.tools.mo", reason="Ticket - 157136")

from common.layer_test_class import check_ir_version
from common.onnx_layer_test_class import OnnxRuntimeLayerTest, onnx_make_model

Expand Down
2 changes: 2 additions & 0 deletions tests/layer_tests/onnx_tests/test_gather.py
Original file line number Diff line number Diff line change
Expand Up @@ -3,6 +3,8 @@

import numpy as np
import pytest
pytest.importorskip("openvino.tools.mo", reason="Ticket - 157136")

from common.layer_test_class import check_ir_version
from common.onnx_layer_test_class import OnnxRuntimeLayerTest, onnx_make_model

Expand Down
2 changes: 2 additions & 0 deletions tests/layer_tests/onnx_tests/test_hard_sigmoid.py
Original file line number Diff line number Diff line change
Expand Up @@ -3,6 +3,8 @@

import numpy as np
import pytest
pytest.importorskip("openvino.tools.mo", reason="Ticket - 157136")

from common.layer_test_class import check_ir_version
from common.onnx_layer_test_class import OnnxRuntimeLayerTest, onnx_make_model

Expand Down
2 changes: 2 additions & 0 deletions tests/layer_tests/onnx_tests/test_identity.py
Original file line number Diff line number Diff line change
Expand Up @@ -3,6 +3,8 @@

import numpy as np
import pytest
pytest.importorskip("openvino.tools.mo", reason="Ticket - 157136")

from common.layer_test_class import check_ir_version
from common.onnx_layer_test_class import OnnxRuntimeLayerTest, onnx_make_model

Expand Down
2 changes: 2 additions & 0 deletions tests/layer_tests/onnx_tests/test_leaky_relu.py
Original file line number Diff line number Diff line change
Expand Up @@ -2,6 +2,8 @@
# SPDX-License-Identifier: Apache-2.0

import pytest
pytest.importorskip("openvino.tools.mo", reason="Ticket - 157136")

from common.layer_test_class import check_ir_version
from common.onnx_layer_test_class import OnnxRuntimeLayerTest, onnx_make_model

Expand Down
2 changes: 2 additions & 0 deletions tests/layer_tests/onnx_tests/test_log.py
Original file line number Diff line number Diff line change
Expand Up @@ -3,6 +3,8 @@

import numpy as np
import pytest
pytest.importorskip("openvino.tools.mo", reason="Ticket - 157136")

from common.layer_test_class import check_ir_version
from common.onnx_layer_test_class import OnnxRuntimeLayerTest, onnx_make_model

Expand Down
2 changes: 2 additions & 0 deletions tests/layer_tests/onnx_tests/test_logsoftmax.py
Original file line number Diff line number Diff line change
Expand Up @@ -3,6 +3,8 @@

import numpy as np
import pytest
pytest.importorskip("openvino.tools.mo", reason="Ticket - 157136")

from common.layer_test_class import check_ir_version
from common.onnx_layer_test_class import OnnxRuntimeLayerTest, onnx_make_model

Expand Down
2 changes: 2 additions & 0 deletions tests/layer_tests/onnx_tests/test_lrn.py
Original file line number Diff line number Diff line change
Expand Up @@ -2,6 +2,8 @@
# SPDX-License-Identifier: Apache-2.0

import pytest
pytest.importorskip("openvino.tools.mo", reason="Ticket - 157136")

from common.layer_test_class import check_ir_version
from common.onnx_layer_test_class import OnnxRuntimeLayerTest, onnx_make_model

Expand Down
2 changes: 2 additions & 0 deletions tests/layer_tests/onnx_tests/test_neg.py
Original file line number Diff line number Diff line change
Expand Up @@ -2,6 +2,8 @@
# SPDX-License-Identifier: Apache-2.0

import pytest
pytest.importorskip("openvino.tools.mo", reason="Ticket - 157136")

from common.layer_test_class import check_ir_version
from common.onnx_layer_test_class import OnnxRuntimeLayerTest, onnx_make_model

Expand Down
2 changes: 2 additions & 0 deletions tests/layer_tests/onnx_tests/test_non_zero.py
Original file line number Diff line number Diff line change
Expand Up @@ -3,6 +3,8 @@

import numpy as np
import pytest
pytest.importorskip("openvino.tools.mo", reason="Ticket - 157136")

from common.layer_test_class import check_ir_version
from common.onnx_layer_test_class import OnnxRuntimeLayerTest, onnx_make_model

Expand Down
2 changes: 2 additions & 0 deletions tests/layer_tests/onnx_tests/test_not.py
Original file line number Diff line number Diff line change
Expand Up @@ -3,6 +3,8 @@

import numpy as np
import pytest
pytest.importorskip("openvino.tools.mo", reason="Ticket - 157136")

from common.layer_test_class import check_ir_version
from common.onnx_layer_test_class import OnnxRuntimeLayerTest, onnx_make_model

Expand Down
2 changes: 2 additions & 0 deletions tests/layer_tests/onnx_tests/test_or.py
Original file line number Diff line number Diff line change
Expand Up @@ -3,6 +3,8 @@

import numpy as np
import pytest
pytest.importorskip("openvino.tools.mo", reason="Ticket - 157136")

from common.layer_test_class import check_ir_version
from common.onnx_layer_test_class import OnnxRuntimeLayerTest, onnx_make_model

Expand Down
2 changes: 2 additions & 0 deletions tests/layer_tests/onnx_tests/test_pad.py
Original file line number Diff line number Diff line change
Expand Up @@ -3,6 +3,8 @@

import numpy as np
import pytest
pytest.importorskip("openvino.tools.mo", reason="Ticket - 157136")

from common.layer_test_class import check_ir_version
from common.onnx_layer_test_class import OnnxRuntimeLayerTest, onnx_make_model

Expand Down
2 changes: 2 additions & 0 deletions tests/layer_tests/onnx_tests/test_pooling.py
Original file line number Diff line number Diff line change
Expand Up @@ -3,6 +3,8 @@

import numpy as np
import pytest
pytest.importorskip("openvino.tools.mo", reason="Ticket - 157136")

from common.layer_test_class import check_ir_version
from common.onnx_layer_test_class import OnnxRuntimeLayerTest, onnx_make_model

Expand Down
2 changes: 2 additions & 0 deletions tests/layer_tests/onnx_tests/test_prelu.py
Original file line number Diff line number Diff line change
Expand Up @@ -3,6 +3,8 @@

import numpy as np
import pytest
pytest.importorskip("openvino.tools.mo", reason="Ticket - 157136")

from common.layer_test_class import check_ir_version
from common.onnx_layer_test_class import OnnxRuntimeLayerTest, onnx_make_model

Expand Down
2 changes: 2 additions & 0 deletions tests/layer_tests/onnx_tests/test_reduce.py
Original file line number Diff line number Diff line change
Expand Up @@ -2,6 +2,8 @@
# SPDX-License-Identifier: Apache-2.0

import pytest
pytest.importorskip("openvino.tools.mo", reason="Ticket - 157136")

from common.layer_test_class import check_ir_version
from common.onnx_layer_test_class import OnnxRuntimeLayerTest, onnx_make_model

Expand Down
2 changes: 2 additions & 0 deletions tests/layer_tests/onnx_tests/test_reduce_lp.py
Original file line number Diff line number Diff line change
Expand Up @@ -5,6 +5,8 @@

import numpy as np
import pytest
pytest.importorskip("openvino.tools.mo", reason="Ticket - 157136")

from common.layer_test_class import check_ir_version
from common.onnx_layer_test_class import OnnxRuntimeLayerTest, onnx_make_model

Expand Down
2 changes: 2 additions & 0 deletions tests/layer_tests/onnx_tests/test_relu.py
Original file line number Diff line number Diff line change
Expand Up @@ -2,6 +2,8 @@
# SPDX-License-Identifier: Apache-2.0

import pytest
pytest.importorskip("openvino.tools.mo", reason="Ticket - 157136")

from common.layer_test_class import check_ir_version
from common.onnx_layer_test_class import OnnxRuntimeLayerTest, onnx_make_model

Expand Down
2 changes: 2 additions & 0 deletions tests/layer_tests/onnx_tests/test_reshape.py
Original file line number Diff line number Diff line change
Expand Up @@ -2,6 +2,8 @@
# SPDX-License-Identifier: Apache-2.0

import pytest
pytest.importorskip("openvino.tools.mo", reason="Ticket - 157136")

from common.layer_test_class import check_ir_version
from common.onnx_layer_test_class import OnnxRuntimeLayerTest, onnx_make_model

Expand Down
2 changes: 2 additions & 0 deletions tests/layer_tests/onnx_tests/test_resize.py
Original file line number Diff line number Diff line change
Expand Up @@ -3,6 +3,8 @@

import numpy as np
import pytest
pytest.importorskip("openvino.tools.mo", reason="Ticket - 157136")

from common.layer_test_class import check_ir_version
from common.onnx_layer_test_class import OnnxRuntimeLayerTest, onnx_make_model

Expand Down
2 changes: 2 additions & 0 deletions tests/layer_tests/onnx_tests/test_roi_align.py
Original file line number Diff line number Diff line change
Expand Up @@ -5,6 +5,8 @@

import numpy as np
import pytest
pytest.importorskip("openvino.tools.mo", reason="Ticket - 157136")

from common.layer_test_class import check_ir_version
from common.onnx_layer_test_class import OnnxRuntimeLayerTest, onnx_make_model
from unit_tests.utils.graph import build_graph
Expand Down
2 changes: 2 additions & 0 deletions tests/layer_tests/onnx_tests/test_scatter.py
Original file line number Diff line number Diff line change
Expand Up @@ -3,6 +3,8 @@

import numpy as np
import pytest
pytest.importorskip("openvino.tools.mo", reason="Ticket - 157136")

from common.layer_test_class import check_ir_version
from common.onnx_layer_test_class import OnnxRuntimeLayerTest, onnx_make_model

Expand Down
2 changes: 2 additions & 0 deletions tests/layer_tests/onnx_tests/test_sigmoid.py
Original file line number Diff line number Diff line change
Expand Up @@ -3,6 +3,8 @@

import numpy as np
import pytest
pytest.importorskip("openvino.tools.mo", reason="Ticket - 157136")

from common.layer_test_class import check_ir_version
from common.onnx_layer_test_class import OnnxRuntimeLayerTest, onnx_make_model

Expand Down
2 changes: 2 additions & 0 deletions tests/layer_tests/onnx_tests/test_sign.py
Original file line number Diff line number Diff line change
Expand Up @@ -2,6 +2,8 @@
# SPDX-License-Identifier: Apache-2.0

import pytest
pytest.importorskip("openvino.tools.mo", reason="Ticket - 157136")

from common.layer_test_class import check_ir_version
from common.onnx_layer_test_class import OnnxRuntimeLayerTest, onnx_make_model

Expand Down
2 changes: 2 additions & 0 deletions tests/layer_tests/onnx_tests/test_softmax.py
Original file line number Diff line number Diff line change
Expand Up @@ -3,6 +3,8 @@

import numpy as np
import pytest
pytest.importorskip("openvino.tools.mo", reason="Ticket - 157136")

from common.layer_test_class import check_ir_version
from common.onnx_layer_test_class import OnnxRuntimeLayerTest, onnx_make_model

Expand Down
2 changes: 2 additions & 0 deletions tests/layer_tests/onnx_tests/test_softplus.py
Original file line number Diff line number Diff line change
Expand Up @@ -2,6 +2,8 @@
# SPDX-License-Identifier: Apache-2.0

import pytest
pytest.importorskip("openvino.tools.mo", reason="Ticket - 157136")

from common.layer_test_class import check_ir_version
from common.onnx_layer_test_class import OnnxRuntimeLayerTest, onnx_make_model

Expand Down
2 changes: 2 additions & 0 deletions tests/layer_tests/onnx_tests/test_softsign.py
Original file line number Diff line number Diff line change
Expand Up @@ -2,6 +2,8 @@
# SPDX-License-Identifier: Apache-2.0

import pytest
pytest.importorskip("openvino.tools.mo", reason="Ticket - 157136")

from common.layer_test_class import check_ir_version
from common.onnx_layer_test_class import OnnxRuntimeLayerTest, onnx_make_model

Expand Down
Loading

0 comments on commit fc4a121

Please sign in to comment.