Skip to content

Commit 029c898

Browse files
committed
Linter fixes
Signed-off-by: Dheeraj Peri <[email protected]>
1 parent aee4739 commit 029c898

File tree

3 files changed

+9
-4
lines changed

3 files changed

+9
-4
lines changed

core/lowering/register_trt_placeholder_ops.cpp

Lines changed: 5 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -10,8 +10,11 @@ c10::AliasAnalysisKind aliasAnalysisFromSchema() {
1010
RegisterOperators trt_placeholder_ops_reg({
1111
/// Op marks a Tensor to be conveted from an Torch Tensor
1212
/// to a TRT constant Tensor
13-
Operator("trt::const(Tensor val) -> Tensor", [](Stack* stack) {}, aliasAnalysisFromSchema()),
13+
Operator(
14+
"trt::const(Tensor val) -> Tensor",
15+
[](Stack* stack) {},
16+
aliasAnalysisFromSchema()),
1417
});
1518

1619
} // namespace jit
17-
} // namespace torch
20+
} // namespace torch

core/plugins/impl/interpolate_plugin.cpp

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -252,7 +252,8 @@ int InterpolatePlugin::enqueue(
252252
cudaStream_t stream) {
253253
#if NV_TENSORRT_MAJOR < 7 || (NV_TENSORRT_MAJOR == 7 && NV_TENSORRT_MINOR < 1)
254254
at::Tensor input = at::from_blob((void*)inputs[0], util::toVec(inputDesc->dims), [](void*) {}, tensor_options_);
255-
at::Tensor output = at::from_blob(outputs[0], util::volume(outputDesc->dims), [](void*) {}, tensor_options_);
255+
at::Tensor output = at::from_blob(
256+
outputs[0], util::volume(outputDesc->dims), [](void*) {}, tensor_options_);
256257

257258
at::cuda::CUDAStream torch_stream = at::cuda::getStreamFromPool();
258259
at::cuda::CUDAStreamGuard torch_guard(torch_stream);

core/plugins/impl/normalize_plugin.cpp

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -184,7 +184,8 @@ int NormalizePlugin::enqueue(
184184
// TRT <= 7.0
185185
#if NV_TENSORRT_MAJOR < 7 || (NV_TENSORRT_MAJOR == 7 && NV_TENSORRT_MINOR < 1)
186186
at::Tensor input = at::from_blob((void*)inputs[0], util::toVec(inputDesc->dims), [](void*) {}, tensor_options_);
187-
at::Tensor output = at::from_blob(outputs[0], util::volume(outputDesc->dims), [](void*) {}, tensor_options_);
187+
at::Tensor output = at::from_blob(
188+
outputs[0], util::volume(outputDesc->dims), [](void*) {}, tensor_options_);
188189

189190
at::cuda::CUDAStream torch_stream = at::cuda::getStreamFromPool();
190191
at::cuda::CUDAStreamGuard torch_guard(torch_stream);

0 commit comments

Comments
 (0)