Skip to content

GitLab

"official/nlp/bert/model_training_utils.py" did not exist on "ab993a21241709398d7cdd1a0b8d7209eec726b4"
Switch branch/tag
Find file BlameHistoryPermalink
  • Bartlomiej Wroblewski's avatar
    Add support for mixed precision in contraction scale and bilinear (#936) · f0748506
    Bartlomiej Wroblewski authored 
    * Extract common functionality to separate files

* Reference contraction: Remove incorrect consts from type_converts

* Reference contraction: Add missing type_convert for dst value

* Reference contraction: Fix incorrect order of B matrix dimensions

* Add support for mixed precision in contraction scale and bilinear

* Move using statements from instances to a common file

* Move using statements from examples to a common file

* Fix the order of B matrix dimensions across examples and profiler

* Fix the computation of error threshold

* Make ComputeDataType an optional argument

* Include possible DataType -> ComputeDataType casting error in the threshold

* Remove commented code
    f0748506
common_instances.hpp 27.3 KB