| 1604 | } |
| 1605 | |
| 1606 | void TensorRTReplacePass::apply(OptState& opt) const { |
| 1607 | if (CompNode::get_device_count(CompNode::DeviceType::CUDA)) { |
| 1608 | opt.set_var_replace_check_flag( |
| 1609 | gopt::VarReplaceCheckFlag::CHECK_SHAPE | |
| 1610 | gopt::VarReplaceCheckFlag::CHECK_DTYPE); |
| 1611 | Impl(*this, opt); |
| 1612 | } else { |
| 1613 | mgb_log_debug("cuda is not available; TensorRTReplacePass is ignored"); |
| 1614 | } |
| 1615 | } |
| 1616 | |
| 1617 | // ===================== TensorRTGraph ================= |
| 1618 | void TensorRTReplacePass::Impl::TensorRTGraph::mark_varnode_format_nchw4() { |