1. 14 2月, 2022 4 次提交
    • C
      [PTen] Add HasAttr for ArgumentMappingContext (#39464) · ddb1e23f
      Chen Weihang 提交于
      * add has_attr for arg map context
      
      * skip useless attr now
      
      * skip attr if not exists
      
      * fix typo
      ddb1e23f
    • C
      [pten] add split kernel (#39060) · d0df5632
      chentianyu03 提交于
      * add split kernel
      
      * add split kernel signature
      
      * fix split bug
      
      * modify MakePtenScalarArrayFromVarList
      
      * modify MakePtenScalarArrayFromVarList
      
      * fix split windows register error
      
      * add test case for split kernel
      
      * replace raw split kernel with pten kernel
      
      * fix makeScalar/ScalarArray bug
      
      * remove debug log
      
      * remove int64_t type in buildPtcontext
      
      * update by code review
      
      * fix split dev test failed
      
      * change DenseTensorMeta to MetaTensor
      
      * change split api code from auto gen to manual
      
      * split cuda kernel support bfloat16 type
      
      * fix conflict
      
      * rm raw split kernel
      
      * merge develop branch
      
      * change to pten::errors
      d0df5632
    • T
      fix gather_nd, *test=kunlun (#39283) · d12c3636
      TTerror 提交于
      d12c3636
    • mhhhh1's avatar
      [MLU] add mlu kernel for c_broadcast op (#39470) · 1b9e6790
      mhhhh1 提交于
      1b9e6790
  2. 11 2月, 2022 11 次提交
  3. 10 2月, 2022 7 次提交
  4. 09 2月, 2022 13 次提交
  5. 08 2月, 2022 5 次提交