1. 15 May, 2019 1 commit
  2. 13 May, 2019 2 commits
  3. 09 May, 2019 1 commit
  4. 30 Apr, 2019 5 commits
  5. 29 Apr, 2019 2 commits
  6. 26 Apr, 2019 1 commit
    • 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
  7. 23 Apr, 2019 1 commit
  8. 18 Apr, 2019 2 commits
  9. 16 Apr, 2019 1 commit
  10. 11 Apr, 2019 1 commit
    • henrymai's avatar
      prelu belongs in FP16_CASTS (#257) · 4dc711bc
      henrymai authored
      The main use of these functions (e.g.: `torch.{conv*, prelu}`) is via their `torch.nn`
      wrapping layers.
      
      The `torch.nn` layers are what contain the weights and call into these lower level
      functions with the weights as a parameter in their `forward()` method.
      
      The `torch.conv*` functions are already in the `FP16_CASTS` list due to amp's philosophy of
      casting the parameters rather than the model/layer weights.
      
      Conceptually `torch.prelu` is the same as the `torch.conv*` case, where its weight parameter
      is passed in from its wrapper layer `torch.nn.PReLU`.
      4dc711bc
  11. 10 Apr, 2019 5 commits
  12. 09 Apr, 2019 1 commit
  13. 08 Apr, 2019 1 commit
  14. 05 Apr, 2019 3 commits
  15. 04 Apr, 2019 3 commits
  16. 03 Apr, 2019 1 commit
  17. 01 Apr, 2019 1 commit
  18. 31 Mar, 2019 1 commit
  19. 27 Mar, 2019 2 commits
  20. 26 Mar, 2019 2 commits
  21. 23 Mar, 2019 1 commit
  22. 22 Mar, 2019 2 commits
    • jjsjann123's avatar
      [SyncBatchNorm] (#206) · 0a991543
      jjsjann123 authored
      supporting 2 dimensional input, resolving issue #194
      
      Implementation:
        for 2d input, switching channel_last flag to true for better memory access
      pattern in the kernel.
      0a991543
    • henrymai's avatar
      Add prelu to list of torch overrides (#217) · 570fde70
      henrymai authored
      * Add prelu to list of torch overrides
      
      This is to fix the following error:
      
        File "/opt/conda/lib/python3.6/site-packages/torch/nn/modules/module.py", line 489, in __call__
          result = self.forward(*input, **kwargs)
        File "/opt/conda/lib/python3.6/site-packages/torch/nn/modules/container.py", line 92, in forward
          input = module(input)
        File "/opt/conda/lib/python3.6/site-packages/torch/nn/modules/module.py", line 489, in __call__
          result = self.forward(*input, **kwargs)
        File "/opt/conda/lib/python3.6/site-packages/torch/nn/modules/activation.py", line 722, in forward
          return F.prelu(input, self.weight)
        File "/opt/conda/lib/python3.6/site-packages/torch/nn/functional.py", line 1040, in prelu
          return torch.prelu(input, weight)
      RuntimeError: expected scalar type Half but found Float
      
      * Update torch_overrides.py
      570fde70