Enable nvComp for SBSA platform (#5889)

- enables building with nvComp for SBSA platform
- enables inflate operator tests for SBSA platform

Signed-off-by: Janusz Lisiecki <[email protected]>

Author: JanuszL

CMakeLists.txt

@@ -143,13 +143,8 @@ else()
   unset(BUILD_CUFILE CACHE)
 endif()
 
-if(NOT (${ARCH} MATCHES "aarch64"))
-  cmake_dependent_option(BUILD_NVCOMP "Build with nvCOMP support" OFF
-                         "NOT BUILD_DALI_NODEPS" OFF)
-else()
-  # make sure that even if set by -DBUILD_NVCOMP, it will be unset as not supported
-  unset(BUILD_NVCOMP CACHE)
-endif()
+cmake_dependent_option(BUILD_NVCOMP "Build with nvCOMP support" OFF
+                        "NOT BUILD_DALI_NODEPS" OFF)
 
 if (BUILD_DALI_NODEPS)
   set(BUILD_OPENCV OFF)

dali/test/python/checkpointing/test_dali_stateless_operators.py

@@ -791,7 +791,7 @@ def numba_pipe():
 
 
 @has_operator("experimental.inflate")
-@restrict_platform(min_compute_cap=6.0, platforms=["x86_64"])
+@restrict_platform(min_compute_cap=6.0)
 @stateless_signed_off("experimental.inflate")
 def test_inflate_stateless():
     import lz4.block

dali/test/python/operator_1/test_inflate.py

@@ -98,7 +98,7 @@ def pipeline():
 
 
 @has_operator("experimental.inflate")
-@restrict_platform(min_compute_cap=6.0, platforms=["x86_64"])
+@restrict_platform(min_compute_cap=6.0)
 def test_sample_inflate():
     seed = 42
     for batch_size in [1, 64, 348]:
@@ -135,7 +135,7 @@ def pipeline():
 
 
 @has_operator("experimental.inflate")
-@restrict_platform(min_compute_cap=6.0, platforms=["x86_64"])
+@restrict_platform(min_compute_cap=6.0)
 def test_scalar_shape():
     largest_prime_smaller_than_2_to_16 = 65521
     prime_larger_than_2_to_16 = 262147
@@ -241,7 +241,7 @@ def pipeline():
 
 
 @has_operator("experimental.inflate")
-@restrict_platform(min_compute_cap=6.0, platforms=["x86_64"])
+@restrict_platform(min_compute_cap=6.0)
 def test_chunks():
     seed = 42
     batch_sizes = [1, 9, 31]
@@ -278,7 +278,7 @@ def test_chunks():
 
 
 @has_operator("experimental.inflate")
-@restrict_platform(min_compute_cap=6.0, platforms=["x86_64"])
+@restrict_platform(min_compute_cap=6.0)
 @params(
     {"chunk_offsets": []},
     {"chunk_sizes": []},
@@ -312,7 +312,7 @@ def _test_validation(pipeline, error_glob, kwargs=None):
 
 
 @has_operator("experimental.inflate")
-@restrict_platform(min_compute_cap=6.0, platforms=["x86_64"])
+@restrict_platform(min_compute_cap=6.0)
 def test_validation():
     @pipeline_def
     def pipeline_2d_shape():

dali/test/python/test_dali_variable_batch_size.py

@@ -1345,7 +1345,7 @@ def video_decoder_pipe(max_batch_size, input_data, device):
 
 
 @has_operator("experimental.inflate")
-@restrict_platform(min_compute_cap=6.0, platforms=["x86_64"])
+@restrict_platform(min_compute_cap=6.0)
 def test_inflate():
     import lz4.block