@Namespace(value="nvinfer1") public static enum nvinfer.BuilderFlag extends Enum<nvinfer.BuilderFlag>
IBuilderConfig::setFlags(), IBuilderConfig::getFlags()
Enum Constant and Description |
---|
kBF16
Enable DataType::kBF16 layer selection, with FP32 fallback.
|
kDEBUG
Enable debugging of layers via synchronizing after every layer.
|
kDIRECT_IO
Require that no reformats be inserted between a layer and a network I/O tensor
for which ITensor::setAllowedFormats was called.
|
kDISABLE_COMPILATION_CACHE
Disable caching of JIT-compilation results during engine build.
|
kDISABLE_TIMING_CACHE
Disable reuse of timing information across identical layers.
|
kERROR_ON_TIMING_CACHE_MISS
Emit error when a tactic being timed is not present in the timing cache.
|
kEXCLUDE_LEAN_RUNTIME
Exclude lean runtime from the plan when version forward compatability is enabled.
|
kFP16
Enable FP16 layer selection, with FP32 fallback.
|
kFP8
Enable plugins with FP8 input/output.
|
kGPU_FALLBACK
Enable layers marked to execute on GPU if layer cannot execute on DLA.
|
kINT4
Enable plugins with INT4 input/output.
|
kINT8
Enable Int8 layer selection, with FP32 fallback with FP16 fallback if kFP16 also specified.
|
kMONITOR_MEMORY
Enable memory monitor during build time.
|
kOBEY_PRECISION_CONSTRAINTS
Require that layers execute in specified precisions.
|
kPREFER_PRECISION_CONSTRAINTS
Prefer that layers execute in specified precisions.
|
kREFIT
Enable building a refittable engine.
|
kREFIT_IDENTICAL
Create a refittable engine under the assumption that the refit weights will be identical to those provided at
build time.
|
kREFIT_INDIVIDUAL
Enable building a refittable engine and provide fine-grained control.
|
kREJECT_EMPTY_ALGORITHMS
Fail if IAlgorithmSelector::selectAlgorithms returns an empty set of algorithms.
|
kSAFETY_SCOPE
Change the allowed parameters in the EngineCapability::kSTANDARD flow to
match the restrictions that EngineCapability::kSAFETY check against for DeviceType::kGPU
and EngineCapability::kDLA_STANDALONE check against the DeviceType::kDLA case.
|
kSPARSE_WEIGHTS
Allow the builder to examine weights and use optimized functions when weights have suitable sparsity.
|
kSTRICT_NANS
Disable floating-point optimizations: 0*x => 0, x-x => 0, or x/x => 1.
|
kSTRIP_PLAN
Strip the refittable weights from the engine plan file.
|
kTF32
Allow (but not require) computations on tensors of type DataType::kFLOAT to use TF32.
|
kVERSION_COMPATIBLE
Restrict to lean runtime operators to provide version forward compatibility
for the plan.
|
kWEIGHT_STREAMING
\brief Enable weight streaming for the current engine.
|
kWEIGHTLESS
Deprecated.
Deprecated in TensorRT 10.0. Superseded by kSTRIP_PLAN.
|
Modifier and Type | Field and Description |
---|---|
int |
value |
Modifier and Type | Method and Description |
---|---|
nvinfer.BuilderFlag |
intern() |
String |
toString() |
static nvinfer.BuilderFlag |
valueOf(String name)
Returns the enum constant of this type with the specified name.
|
static nvinfer.BuilderFlag[] |
values()
Returns an array containing the constants of this enum type, in
the order they are declared.
|
public static final nvinfer.BuilderFlag kFP16
public static final nvinfer.BuilderFlag kINT8
public static final nvinfer.BuilderFlag kDEBUG
public static final nvinfer.BuilderFlag kGPU_FALLBACK
public static final nvinfer.BuilderFlag kREFIT
public static final nvinfer.BuilderFlag kDISABLE_TIMING_CACHE
public static final nvinfer.BuilderFlag kTF32
public static final nvinfer.BuilderFlag kSPARSE_WEIGHTS
public static final nvinfer.BuilderFlag kSAFETY_SCOPE
public static final nvinfer.BuilderFlag kOBEY_PRECISION_CONSTRAINTS
public static final nvinfer.BuilderFlag kPREFER_PRECISION_CONSTRAINTS
public static final nvinfer.BuilderFlag kDIRECT_IO
public static final nvinfer.BuilderFlag kREJECT_EMPTY_ALGORITHMS
public static final nvinfer.BuilderFlag kVERSION_COMPATIBLE
public static final nvinfer.BuilderFlag kEXCLUDE_LEAN_RUNTIME
public static final nvinfer.BuilderFlag kFP8
nvinfer.HardwareCompatibilityLevel
public static final nvinfer.BuilderFlag kERROR_ON_TIMING_CACHE_MISS
public static final nvinfer.BuilderFlag kBF16
public static final nvinfer.BuilderFlag kDISABLE_COMPILATION_CACHE
public static final nvinfer.BuilderFlag kSTRIP_PLAN
public static final nvinfer.BuilderFlag kWEIGHTLESS
public static final nvinfer.BuilderFlag kREFIT_IDENTICAL
public static final nvinfer.BuilderFlag kWEIGHT_STREAMING
IRuntime::deserializeCudaEngine,
ICudaEngine::getMinimumWeightStreamingBudget,
ICudaEngine::setWeightStreamingBudget
public static final nvinfer.BuilderFlag kINT4
public static final nvinfer.BuilderFlag kREFIT_INDIVIDUAL
public static final nvinfer.BuilderFlag kSTRICT_NANS
public static final nvinfer.BuilderFlag kMONITOR_MEMORY
public static nvinfer.BuilderFlag[] values()
for (nvinfer.BuilderFlag c : nvinfer.BuilderFlag.values()) System.out.println(c);
public static nvinfer.BuilderFlag valueOf(String name)
name
- the name of the enum constant to be returned.IllegalArgumentException
- if this enum type has no constant with the specified nameNullPointerException
- if the argument is nullpublic nvinfer.BuilderFlag intern()
public String toString()
toString
in class Enum<nvinfer.BuilderFlag>
Copyright © 2024. All rights reserved.