unity_build_rule.cmake 890 字节
Newer Older
1 2 3 4 5 6
# 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.
7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37
register_unity_group(
  cc
  ftrl_op.cc
  lars_momentum_op.cc
  momentum_op.cc
  sgd_op.cc
  proximal_adagrad_op.cc
  adagrad_op.cc
  adam_op.cc
  adamax_op.cc
  dgc_momentum_op.cc
  proximal_gd_op.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
  proximal_adagrad_op.cu
  adagrad_op.cu
  adam_op.cu
  adamax_op.cu
  decayed_adagrad_op.cu
  adadelta_op.cu
  lamb_op.cu
  rmsprop_op.cu)