d86de6c98e
A new `env.Run` method is added which allows to control the verbosity of builders output automatically depending on whether the "verbose" option is set. It also allows to optionally run any SCons commands in a subprocess using the existing `run_in_subprocess` method, unifying the interface. `Action` objects wrap all builder functions to include a short build message associated with any action. Notably, this removes quite verbose output generated by `make_doc_header` and `make_editor_icons_action` builders.
118 lines
4.2 KiB
Python
118 lines
4.2 KiB
Python
#!/usr/bin/env python
|
|
|
|
import resource_to_cpp
|
|
|
|
Import("env")
|
|
Import("env_modules")
|
|
|
|
env_oidn = env_modules.Clone()
|
|
|
|
# Thirdparty source files
|
|
thirdparty_dir = "#thirdparty/oidn/"
|
|
thirdparty_sources = [
|
|
"core/api.cpp",
|
|
"core/device.cpp",
|
|
"core/filter.cpp",
|
|
"core/network.cpp",
|
|
"core/autoencoder.cpp",
|
|
"core/transfer_function.cpp",
|
|
"weights/rtlightmap_hdr.gen.cpp",
|
|
"mkl-dnn/src/common/batch_normalization.cpp",
|
|
"mkl-dnn/src/common/concat.cpp",
|
|
"mkl-dnn/src/common/convolution.cpp",
|
|
"mkl-dnn/src/common/convolution_pd.cpp",
|
|
"mkl-dnn/src/common/deconvolution.cpp",
|
|
"mkl-dnn/src/common/eltwise.cpp",
|
|
"mkl-dnn/src/common/engine.cpp",
|
|
"mkl-dnn/src/common/inner_product.cpp",
|
|
"mkl-dnn/src/common/inner_product_pd.cpp",
|
|
"mkl-dnn/src/common/lrn.cpp",
|
|
"mkl-dnn/src/common/memory.cpp",
|
|
"mkl-dnn/src/common/memory_desc_wrapper.cpp",
|
|
"mkl-dnn/src/common/mkldnn_debug.cpp",
|
|
"mkl-dnn/src/common/mkldnn_debug_autogenerated.cpp",
|
|
"mkl-dnn/src/common/pooling.cpp",
|
|
"mkl-dnn/src/common/primitive.cpp",
|
|
"mkl-dnn/src/common/primitive_attr.cpp",
|
|
"mkl-dnn/src/common/primitive_desc.cpp",
|
|
"mkl-dnn/src/common/primitive_exec_types.cpp",
|
|
"mkl-dnn/src/common/primitive_iterator.cpp",
|
|
"mkl-dnn/src/common/query.cpp",
|
|
"mkl-dnn/src/common/reorder.cpp",
|
|
"mkl-dnn/src/common/rnn.cpp",
|
|
"mkl-dnn/src/common/scratchpad.cpp",
|
|
"mkl-dnn/src/common/shuffle.cpp",
|
|
"mkl-dnn/src/common/softmax.cpp",
|
|
"mkl-dnn/src/common/stream.cpp",
|
|
"mkl-dnn/src/common/sum.cpp",
|
|
"mkl-dnn/src/common/utils.cpp",
|
|
"mkl-dnn/src/common/verbose.cpp",
|
|
"mkl-dnn/src/cpu/cpu_barrier.cpp",
|
|
"mkl-dnn/src/cpu/cpu_concat.cpp",
|
|
"mkl-dnn/src/cpu/cpu_engine.cpp",
|
|
"mkl-dnn/src/cpu/cpu_memory.cpp",
|
|
"mkl-dnn/src/cpu/cpu_reducer.cpp",
|
|
"mkl-dnn/src/cpu/cpu_reorder.cpp",
|
|
"mkl-dnn/src/cpu/cpu_sum.cpp",
|
|
"mkl-dnn/src/cpu/jit_avx2_conv_kernel_f32.cpp",
|
|
"mkl-dnn/src/cpu/jit_avx2_convolution.cpp",
|
|
"mkl-dnn/src/cpu/jit_avx512_common_conv_kernel.cpp",
|
|
"mkl-dnn/src/cpu/jit_avx512_common_conv_winograd_kernel_f32.cpp",
|
|
"mkl-dnn/src/cpu/jit_avx512_common_convolution.cpp",
|
|
"mkl-dnn/src/cpu/jit_avx512_common_convolution_winograd.cpp",
|
|
"mkl-dnn/src/cpu/jit_avx512_core_fp32_wino_conv_2x3.cpp",
|
|
"mkl-dnn/src/cpu/jit_avx512_core_fp32_wino_conv_4x3.cpp",
|
|
"mkl-dnn/src/cpu/jit_avx512_core_fp32_wino_conv_4x3_kernel.cpp",
|
|
"mkl-dnn/src/cpu/jit_sse42_conv_kernel_f32.cpp",
|
|
"mkl-dnn/src/cpu/jit_sse42_convolution.cpp",
|
|
"mkl-dnn/src/cpu/jit_transpose_src_utils.cpp",
|
|
"mkl-dnn/src/cpu/jit_uni_eltwise.cpp",
|
|
"mkl-dnn/src/cpu/jit_uni_pool_kernel_f32.cpp",
|
|
"mkl-dnn/src/cpu/jit_uni_pooling.cpp",
|
|
"mkl-dnn/src/cpu/jit_uni_reorder.cpp",
|
|
"mkl-dnn/src/cpu/jit_uni_reorder_utils.cpp",
|
|
"mkl-dnn/src/cpu/jit_utils/jit_utils.cpp",
|
|
"mkl-dnn/src/cpu/jit_utils/jitprofiling/jitprofiling.c",
|
|
"common/platform.cpp",
|
|
"common/thread.cpp",
|
|
"common/tensor.cpp",
|
|
]
|
|
thirdparty_sources = [thirdparty_dir + file for file in thirdparty_sources]
|
|
|
|
thirdparty_include_dirs = [
|
|
"",
|
|
"include",
|
|
"mkl-dnn/include",
|
|
"mkl-dnn/src",
|
|
"mkl-dnn/src/common",
|
|
"mkl-dnn/src/cpu/xbyak",
|
|
"mkl-dnn/src/cpu",
|
|
]
|
|
thirdparty_include_dirs = [thirdparty_dir + file for file in thirdparty_include_dirs]
|
|
|
|
|
|
env_oidn.Prepend(CPPPATH=thirdparty_include_dirs)
|
|
env_oidn.Append(
|
|
CPPDEFINES=[
|
|
"MKLDNN_THR=MKLDNN_THR_SEQ",
|
|
"OIDN_STATIC_LIB",
|
|
"__STDC_CONSTANT_MACROS",
|
|
"__STDC_LIMIT_MACROS",
|
|
"DISABLE_VERBOSE",
|
|
"MKLDNN_ENABLE_CONCURRENT_EXEC",
|
|
"NDEBUG",
|
|
]
|
|
)
|
|
|
|
env_thirdparty = env_oidn.Clone()
|
|
env_thirdparty.disable_warnings()
|
|
env_thirdparty.add_source_files(env.modules_sources, thirdparty_sources)
|
|
|
|
weights_in_path = thirdparty_dir + "weights/rtlightmap_hdr.tza"
|
|
weights_out_path = thirdparty_dir + "weights/rtlightmap_hdr.gen.cpp"
|
|
|
|
env_thirdparty.Depends(weights_out_path, weights_in_path)
|
|
env_thirdparty.CommandNoCache(weights_out_path, weights_in_path, resource_to_cpp.tza_to_cpp)
|
|
|
|
env_oidn.add_source_files(env.modules_sources, "denoise_wrapper.cpp")
|
|
env_modules.add_source_files(env.modules_sources, ["register_types.cpp", "lightmap_denoiser.cpp"])
|