1. 16 2月, 2023 1 次提交
    • Z
      Add mean composite rule (#50298) · f7f67b72
      zqw_1997 提交于
      * beta
      
      * small commit
      
      * add batch_norm composite rule
      
      move composite test case
      
      remove unuseful var
      
      add composite op blacklist
      
      * small change v2
      
      * finish the test_composite_mean and test_composite_mean_grad
      
      * add ops assertion to the tests
      
      * add cinn test
      
      * fix the error and inappropriate usage in func: mean_composite
      
      * remove the ref of outer lib in primtives.py
      
      * modify sample code of reduce_sum
      
      * fix composite mean op map
      
      * modify testcases to test more float type
      
      * remove cpu float16 test
      
      * cinn test fix
      
      * remove reduce_max
      
      * change the name sum to sum_x
      
      * change the use of reduce_sum to sum
      
      ---------
      Co-authored-by: Ncyber-pioneer <chenzhuo@tju.edu.cn>
      f7f67b72
  2. 15 2月, 2023 1 次提交
    • C
      fix composite op map (#50397) · ff86aeab
      cyber-pioneer 提交于
      * map output from composite rule to origin op
      
      add mean layer_norm dropout op map
      
      add input map check
      
      composite softmax support input shape []
      
      * composite softmax support shape []
      
      * polish log
      
      * solve conflict
      
      * polish code
      
      * polish op map output
      
      * add check dtype
      ff86aeab
  3. 14 2月, 2023 2 次提交
    • M
      b85af464
    • G
      Add gelu composite rule (#50295) · c364f41d
      GGBond8488 提交于
      * add gelu composite rule
      
      * use full replace fill_constant
      
      * change the form of calculation
      
      * remove float16 test for composite gelu
      
      * reformate code
      
      * remove float16 test case
      
      * add forwad with prim and backward without prim test
      
      * add float16 test for composite gelu and add high dims test
      
      * add float16 test case and high dims test
      
      * shield float16 and cpu test case
      
      * increase train step to 10 in test cinn prim gelu
      
      * replace pow to multiply
      c364f41d
  4. 09 2月, 2023 1 次提交
  5. 07 2月, 2023 1 次提交
  6. 17 1月, 2023 1 次提交
  7. 13 1月, 2023 1 次提交