• Adam Osewski's avatar
    Softmax unit-test reduction across all and non innermost dims cases. (#406) · 6d8614ee
    Adam Osewski authored
    
    
    * Add reduction across all dims cases.
    
    * host softmax: handle all reduce
    
    * Test cases when reduced dim is not innermost axis.
    
    * Fix syntax.
    
    * Test non innermost dim for fp32 and int8
    
    * Group test suites wrt NumReduceDim.
    
    * Additionally test failing cases.
    
    * Throw error when Rank or NumReduceDims doesn't match arguments.
    
    * Check reducedDims has correct values
    
    * Move don't reuse DeviceReduceMultiblock IsSupportedArgument method.
    Instead implement own. (in fact just get rid of one check to enable
    reduction across inner dimensions).
    
    * Reorganize unit tests to better cover use scenarios.
    
    * Test input validation
    * Test reduction of inner dimensions with custom op instances.
    
    * Refactor fp32 and int8 unit tests.
    
    * Fix FP32 instance template parameters.
    
    * Add more instances.
    
    * Instances with InSrcVectorDim=0.
    
    * Do not initialize and copy data when arg not supported.
    
    * ckProfiler Softmax use instance factory.
    
    * Refactor device softmax IsSupported.
    
    * Additionally add non-polymorphic api functions
    
    * Split softmax instances into multiple files.
    
    * Fix profiler.
    
    * Reorganize tests to reuse profiler and cover edge cases.
    
    * Clang-format
    
    * I8 Softmax instances along with UT.
    
    * Reuse type alias definitions from instance factory header.
    
    * Clean included headers
    
    * Fix variable names.
    
    * Add missing checks in Argument constructor.
    Co-authored-by: default avatarAdam Osewski <aosewski@amd.com>
    Co-authored-by: default avatarAnthony Chang <ac.chang@outlook.com>
    6d8614ee
profiler.cpp 5 KB