Skip to content

GitLab

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
multi_tensor_scale_kernel.cu 3 KB