Skip to content

GitLab

"README_ORINGIN.md" did not exist on "c38a7d7ddd9c612e368adec1ef94583be602fc7e"
Switch branch/tag
Find file BlameHistoryPermalink
  • ptrblck's avatar
    Replace type().ScalarType() with scalar_type() (#272) · 855808f3
    ptrblck authored 
    * change .type().ScalarType() to .scalar_type() + at::ScalarType::X to at::kX

* revert scalar_type() to type() for AT_DISPATCH_FLOATING_TYPES_AND_HALF

* revert scalar_type() to type() in AT_DISPATCH_FLOATING_TYPES

* revert scalar_type() to type() for AT_DISPATCH_FLOATING_TYPES_AND_HALF in welford.cu

* revert scalar_type() to type() in layer_norm_cuda_kernel.cu

* revert at::kType  to at::ScalarType::Type

* use DISPATCH_FLOAT_AND_HALF to get rid of warnings

* add dispatch mechanisms for double+float and double+float+half
    855808f3
layer_norm_cuda_kernel.cu 23.9 KB