• C
    [PTen] Add cmake function for kernels (#38311) · e6310dbd
    Chen Weihang 提交于
    * add pten kernel cmake
    
    * add pten kernel cmake function
    
    * fix compile error
    
    * add enforce include for full kernel
    
    * fix compile failed
    
    * change cuda to gpu
    
    * fix cmake function error
    e6310dbd
flatten_kernel.cc 4.1 KB