未验证 提交 55704db5 编写于 作者: B bukejiyu 提交者: GitHub

[inference][trt]test_emb_eltwise_layernorm_fuse_pass cuda12 fix (#54757)

* modify tensorrt ci timeout

* activation ci bug fix

* Update CMakeLists.txt

* Update CMakeLists.txt

* comment out  int8 mode test_trt_dynamic_shape_groupnorm

* fix test_sync_batch_norm_op Segmentation fault
上级 e9f8baa6
......@@ -102,7 +102,8 @@ class ElementwiseTensorOpConverter : public OpConverter {
int right_one_num = dims_x.nbDims - axis - dims_y.nbDims;
nvinfer1::IShuffleLayer* reshape_layer;
nvinfer1::ITensor* reshape_y_tensor;
if (left_one_num > 0 || right_one_num > 0) {
if (dims_x.nbDims != dims_y.nbDims &&
(left_one_num > 0 || right_one_num > 0)) {
if (engine_->with_dynamic_shape()) {
auto* y_shape_tensor = Shape(Y);
auto* new_y_shape_tensor = y_shape_tensor;
......
......@@ -2567,7 +2567,7 @@ struct SimpleOpTypeSetTeller : public Teller {
return false;
}
}
if (op_type == "lookup_table") {
if (op_type == "lookup_table" || op_type == "lookup_table_v2") {
if (!with_dynamic_shape) {
VLOG(3) << "the lookup_table does not support "
"static shape yet";
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册