unity_build_rule.cmake

# This file records the Unity Build compilation rules.
# The source files in a `register_unity_group` called are compiled in a unity
# file.
# Generally, the combination rules in this file do not need to be modified.
# If there are some redefined error in compiling with the source file which
# in combination rule, you can remove the source file from the following rules.
register_unity_group(cc
    ftrl_op.cc
    lars_momentum_op.cc
    momentum_op.cc
    sgd_op.cc)
register_unity_group(cc
    adagrad_op.cc
    adam_op.cc
    adamax_op.cc
    dgc_momentum_op.cc
    proximal_gd_op.cc)
register_unity_group(cc
    decayed_adagrad_op.cc
    adadelta_op.cc
    lamb_op.cc
    dpsgd_op.cc
    rmsprop_op.cc)
register_unity_group(cu
    ftrl_op.cu
    lars_momentum_op.cu
    momentum_op.cu
    sgd_op.cu)
register_unity_group(cu
    adagrad_op.cu
    adam_op.cu
    adamax_op.cu)
register_unity_group(cu
    decayed_adagrad_op.cu
    adadelta_op.cu
    lamb_op.cu
    rmsprop_op.cu)
# The following groups are to make better use of `/MP` which MSVC's parallel
# compilation instruction when compiling in Unity Build.
register_unity_group(cu proximal_adagrad_op.cu)