@@ -11,8 +11,8 @@ namespace trtorch {
11
11
namespace core {
12
12
namespace conversion {
13
13
14
- // Defined in core/conversion/conversion_blacklist .cpp
15
- bool isNodeConversionBlacklisted (const torch::jit::Node* n);
14
+ // Defined in core/conversion/conversion_ignorelist .cpp
15
+ bool isNodeConversionIgnored (const torch::jit::Node* n);
16
16
17
17
bool OpSupported (const torch::jit::Node* n) {
18
18
return evaluators::shouldEvalAtConversionTime (n) || converters::node_is_convertable (n);
@@ -302,7 +302,7 @@ void ConvertBlockToNetDef(ConversionCtx* ctx, const torch::jit::Block* b, Conver
302
302
303
303
for (const auto n : nodes) {
304
304
bool to_eval = evaluators::shouldEvalAtConversionTime (n);
305
- bool blacklisted = isNodeConversionBlacklisted (n);
305
+ bool ignored = isNodeConversionIgnored (n);
306
306
if (n->kind () == torch::jit::prim::Loop) {
307
307
EvaluateLoopBlock (ctx, n);
308
308
} else if (n->kind () == torch::jit::prim::If) {
@@ -317,16 +317,16 @@ void ConvertBlockToNetDef(ConversionCtx* ctx, const torch::jit::Block* b, Conver
317
317
}
318
318
ctx->AssociateValueAndIValue (n->output (0 ), eval.value ());
319
319
}
320
- } else if (!blacklisted ) {
320
+ } else if (!ignored ) {
321
321
// Should error out if something fails
322
322
AddLayer (ctx, n);
323
323
} else {
324
324
std::string reason = " " ;
325
325
if (to_eval) {
326
326
reason += " (to be evaluated)" ;
327
327
}
328
- if (blacklisted ) {
329
- reason += " (explicitly blacklisted )" ;
328
+ if (ignored ) {
329
+ reason += " (explicitly ignored )" ;
330
330
}
331
331
LOG_DEBUG (ctx->logger ,
332
332
" Skipping Node: " << util::node_info (n) << reason);
0 commit comments