Fix CMake

Signed-off-by: Rafal Banas <rbanas@nvidia.com>
NVIDIA · Jul 28, 2023 · 449348b · 449348b
1 parent e32a0fd
commit 449348b
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/dali/operators/image/CMakeLists.txt b/dali/operators/image/CMakeLists.txt
@@ -18,7 +18,7 @@ add_subdirectory(color)
 add_subdirectory(crop)
 add_subdirectory(convolution)
 add_subdirectory(distortion)
-if ($BUILD_CVCUDA)
+if (${BUILD_CVCUDA})
     add_subdirectory(filter)
 endif()
 add_subdirectory(mask)

diff --git a/dali/operators/image/filter/median_blur.cc b/dali/operators/image/filter/median_blur.cc
@@ -33,6 +33,7 @@ Median blur performs smoothing of an image or sequence of images by calculating
 the median color of surrounding rectangular region for each pixel.
   )doc")
   .NumInput(1)
+  .InputDox(0, "input", "TensorList",
             "Input data. Must be images in HWC or CHW layout, or a sequence of those.")
   .NumOutput(1)
   .InputLayout({"HWC", "FHWC", "CHW", "FCHW"})