Skip to content

Add a rule to build the examples with STF #1

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
caugonnet wants to merge 2 commits into
base: cudastf
Choose a base branch
from
Open

Add a rule to build the examples with STF #1

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
2 commits
Select commit Hold shift + click to select a range
ca64751
Add a rrule to build the examples with STF too
caugonnet Jan 9, 2025
dc7f8ab
Minor cleanups
caugonnet Jan 22, 2025
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
18 changes: 15 additions & 3 deletions examples/CMakeLists.txt
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -20,9 +20,6 @@ set(examples
black_scholes
print_styles)




add_library(example_lib INTERFACE)
target_include_directories(example_lib SYSTEM INTERFACE ${CUTLASS_INC} ${pybind11_INCLUDE_DIR} ${PYTHON_INCLUDE_DIRS})

Expand Down Expand Up @@ -54,6 +51,21 @@ foreach( example ${examples} )
target_link_libraries(${example} example_lib)
endforeach()

# Compile all examples with CUDASTF and append their name with _stf
foreach( example ${examples} )
string( CONCAT file ${example} ".cu" )

set(output_name "${example}_stf")
add_executable( ${output_name} ${file} )

# Add the -DUSE_STF compilation flag
target_compile_definitions(${output_name} PRIVATE USE_STF)

target_link_libraries(${output_name} example_lib)
endforeach()



# Build proprietary examples
file (GLOB_RECURSE proprietary_sources CONFIGURE_DEPENDS ${CMAKE_SOURCE_DIR}/proprietary/*/examples/*.cu)
foreach (pexample ${proprietary_sources})
Expand Down
4 changes: 0 additions & 4 deletions examples/simple_radar_pipeline.h
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -35,10 +35,6 @@
#include <memory>
#include <stdint.h>

#ifndef USE_STF
#define USE_STF 1
#endif

using namespace matx;

/**
Expand Down