You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
if (n->kind() == torch::jit::prim::Loop || n->kind() == torch::jit::prim::If ||
440
+
converters::node_is_convertable(n)) {
440
441
if (n->blocks().size() > 0) {
441
442
for (constauto sub_b : n->blocks()) {
442
443
auto sub_b_convertable_ops = ConvertableOpsInBlock(sub_b);
@@ -446,8 +447,7 @@ std::set<std::string> ConvertableOpsInBlock(const torch::jit::Block* b) {
446
447
if (converters::node_is_convertable(n)) {
447
448
auto schema = n->maybeSchema();
448
449
TRTORCH_CHECK(
449
-
schema,
450
-
"Unable to get schema for Node " << util::node_info(n) << " (conversion.CheckForConvertableOps)");
450
+
schema, "Unable to get schema for Node " << util::node_info(n) << " (conversion.CheckForConvertableOps)");
451
451
std::stringstream ss;
452
452
ss << *schema;
453
453
convertable_ops.insert(ss.str());
@@ -476,10 +476,12 @@ bool VerifyConverterSupportForBlock(const torch::jit::Block* b) {
476
476
477
477
if (ConvertableOpsInBlock(b).size() == 0) {
478
478
std::stringstream unsupported_msg;
479
-
unsupported_msg << "Method requested cannot be compiled by TRTorch.\nThere is no work to be done since the resulting compiled program will contain an engine that is empty."
480
-
<< std::endl;
481
-
unsupported_msg << "This may be because there are no operators that can be added to the TensorRT graph or all operators have a resolved compile time value."
482
-
<< std::endl;
479
+
unsupported_msg
480
+
<< "Method requested cannot be compiled by TRTorch.\nThere is no work to be done since the resulting compiled program will contain an engine that is empty."
481
+
<< std::endl;
482
+
unsupported_msg
483
+
<< "This may be because there are no operators that can be added to the TensorRT graph or all operators have a resolved compile time value."
0 commit comments