Skip to content

Commit

Permalink
Minor updates
Browse files Browse the repository at this point in the history
  • Loading branch information
mhaseeb123 committed Oct 4, 2024
1 parent 74763b0 commit 06afb49
Show file tree
Hide file tree
Showing 3 changed files with 8 additions and 5 deletions.
1 change: 1 addition & 0 deletions cpp/examples/parquet_io/CMakeLists.txt
Original file line number Diff line number Diff line change
Expand Up @@ -17,6 +17,7 @@ project(
include(../fetch_dependencies.cmake)

add_library(parquet_io_utils OBJECT common_utils.cpp io_source.cpp)
target_compile_features(parquet_io_utils PRIVATE cxx_std_17)
target_link_libraries(parquet_io_utils PRIVATE cudf::cudf)

# Build and install parquet_io
Expand Down
1 change: 1 addition & 0 deletions cpp/examples/parquet_io/io_source.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -25,6 +25,7 @@
#include <thrust/host_vector.h>

#include <filesystem>
#include <fstream>
#include <string>

rmm::host_async_resource_ref pinned_memory_resource()
Expand Down
11 changes: 6 additions & 5 deletions cpp/examples/parquet_io/parquet_io_multithreaded.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -30,6 +30,7 @@
#include <filesystem>
#include <stdexcept>
#include <string>
#include <thread>

/**
* @file parquet_io_multithreaded.cpp
Expand Down Expand Up @@ -349,11 +350,11 @@ int32_t main(int argc, char const** argv)
input_multiplier = std::max(input_multiplier, std::stoi(std::string{argv[2]}));
[[fallthrough]];
case 2:
if (auto arg = std::string{argv[1]}; arg == "-h" or arg == "--help") {
print_usage();
return 0;
} else
input_paths = std::string{argv[1]};
// Check if instead of input_paths, the first argument is `-h` or `--help`
if (auto arg = std::string{argv[1]}; arg != "-h" and arg != "--help") {
input_paths = std::move(arg);
break;
}
[[fallthrough]];
default: print_usage(); throw std::runtime_error("");
}
Expand Down

0 comments on commit 06afb49

Please sign in to comment.