Merge pull request #599 from NVIDIA/jetpack_test_fix

fix(//tests): fix optional jetson tests

Author: narendasan

tests/BUILD

@@ -18,7 +18,7 @@ test_suite(
     name = "aarch64_tests",
     tests = [
         "//tests/core:core_tests",
-        "//tests/modules:aarch64_module_tests",
+        "//tests/cpp:aarch64_api_tests",
     ],
 )
 

tests/accuracy/test_dla_fp16_accuracy.cpp

@@ -27,7 +27,7 @@ TEST_P(AccuracyTests, DLAFP16AccuracyIsClose) {
 
   std::vector<std::vector<int64_t>> input_shape = {{32, 3, 32, 32}};
   auto compile_spec = trtorch::CompileSpec({input_shape});
-  compile_spec.enabled_precisions.insert(torch::kF16);
+  compile_spec.enabled_precisions = {torch::kF16};
   compile_spec.device.device_type = trtorch::CompileSpec::Device::DeviceType::kDLA;
   compile_spec.device.gpu_id = 0;
   compile_spec.device.dla_core = 1;

tests/accuracy/test_dla_int8_accuracy.cpp

@@ -21,7 +21,7 @@ TEST_P(AccuracyTests, DLAINT8AccuracyIsClose) {
   // Configure settings for compilation
   auto compile_spec = trtorch::CompileSpec({input_shape});
   // Set operating precision to INT8
-  compile_spec.enabled_precisions.insert(torch::kI8);
+  compile_spec.enabled_precisions = {torch::kF16, torch::kI8};
   // Use the TensorRT Entropy Calibrator
   compile_spec.ptq_calibrator = calibrator;
   // Set max batch size for the engine