CARVIEW |
Navigation Menu
-
Notifications
You must be signed in to change notification settings - Fork 5.8k
【Paddle TensorRT】pir-trt support roi_align #71041
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Conversation
python/paddle/tensorrt/impls/ops.py
Outdated
spatial_scale = paddle_op.attrs().get("spatial_scale") | ||
sampling_ratio = paddle_op.attrs().get("sampling_ratio") | ||
aligned = paddle_op.attrs().get("aligned") | ||
type_id = int(WithFp16()) |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
这里的Fp16是全图的属性吗?
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
是的
你的PR提交成功,感谢你对开源项目的贡献! |
auto pooled_height_attr = | ||
op->attribute<pir::Int32Attribute>("pooled_height").data(); | ||
if (pooled_height_attr <= 0) { | ||
return false; |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
这里加一个VLOG
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
done
auto pooled_width_attr = | ||
op->attribute<pir::Int32Attribute>("pooled_width").data(); | ||
if (pooled_width_attr <= 0) { | ||
return false; |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
同上
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
done
auto spatial_scale_attr = | ||
op->attribute<pir::FloatAttribute>("spatial_scale").data(); | ||
if (spatial_scale_attr <= 0.f) { | ||
return false; |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
同上
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
done
python/paddle/tensorrt/util.py
Outdated
@@ -223,6 +223,8 @@ def __new__(cls, trt_config=None): | |||
if not cls._instance: | |||
cls._instance = super().__new__(cls) | |||
cls._instance.trt_config = trt_config | |||
elif trt_config is not None: |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
加这个是干什么的
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
done
5d6a4f3
to
456a4f9
Compare
* pir roi_align * update
* pir roi_align * update
PR Category
Inference
PR Types
Others
Description
pcard-71500
pir-convert: roi_align