1. 07 Oct, 2022 2 commits
  2. 04 Oct, 2022 2 commits
  3. 03 Oct, 2022 1 commit
    • Umang Yadav's avatar
      Add output_alias and runs_on_offload_target flags for the custom ops (#1309) · c9ffb38d
      Umang Yadav authored
      Adds two methods for the custom_ops virtual class.
      
      bool runs_on_offload_target(), if the custom op runs directly on the gpu then it should be set to true. in this case, custom op expects its parameters to reside in GPU memory and writes output to the GPU memory. If it is set to false then, custom op expects it's parameter to reside on the host and puts back the result into the host memory.
      
      output_alias, if output of the custom op is aliasing the input buffer. i.e. interpreting the same input buffer with differnet shape and strides.
      
      Update as_vector() in C++ API to handle non-standard shapes. It required exposing element_index to space_index conversion method for the shape class.
      c9ffb38d
  4. 29 Sep, 2022 2 commits
  5. 28 Sep, 2022 1 commit
    • Umang Yadav's avatar
      Add compute_fp32 flag for quant_gemm tests (#1360) · 70e63960
      Umang Yadav authored
      test_gpu_pack_int8_args fails on gfx908 machine, because it doesn't set compute_fp32 flag correctly. This PR fixes the test such that it checks for the device-name, and rocblas-versions and sets this flag accordingly.
      70e63960
  6. 27 Sep, 2022 1 commit
  7. 26 Sep, 2022 3 commits
  8. 23 Sep, 2022 1 commit
  9. 21 Sep, 2022 2 commits
  10. 19 Sep, 2022 1 commit
    • Paul Fultz II's avatar
      Improve layernorm and reductions performance (#1348) · 97a1ed2d
      Paul Fultz II authored
      Compute mean and variance in same reduction
      Set block size to numbers divisible by 32 instead powers of 2
      Global is also set exactly instead of being divisible by block size
      More exact matching of global/local can help get rid of branching/loops
      Reduce vectors first before doing dpp_reduce
      Explicitly vectorize array operators since the compiler doesnt always vectorize them
      Still uses old for loop when its computing at compile-time since the reinterpret_cast nor the all the vector types is supported
      97a1ed2d
  11. 16 Sep, 2022 2 commits
  12. 15 Sep, 2022 1 commit
  13. 14 Sep, 2022 3 commits
  14. 13 Sep, 2022 1 commit
    • turneram's avatar
      Use rocblas_gemm_ex for batched gemms with broadcasted B (#1354) · a10a8ef1
      turneram authored
      Improves performance for 4/6 GEMMs used by huggingface BERT models with batch_size>1 by using a non-batched rocBLAS call for GEMMs where the B input has a broadcasted batch dimension.
      The four verify tests added reflect the actual configurations used by bert-base-cased, with varied batch sizes.
      
      Also adds a matcher to simplify_reshapes to move multibroadcasts after concats.
      a10a8ef1
  15. 08 Sep, 2022 2 commits
  16. 07 Sep, 2022 1 commit
  17. 06 Sep, 2022 1 commit
  18. 31 Aug, 2022 1 commit
  19. 29 Aug, 2022 1 commit
  20. 27 Aug, 2022 2 commits
  21. 26 Aug, 2022 1 commit
  22. 24 Aug, 2022 1 commit
  23. 23 Aug, 2022 1 commit
    • Charlie Lin's avatar
      Dynamic ref NMS (#1288) · fa3c21fa
      Charlie Lin authored
      Has NMS op output a dynamic shape (ONNX spec behavior)
      Allows for dynamic input shape to NMS op
      fa3c21fa
  24. 21 Aug, 2022 1 commit
    • varunsh's avatar
      Update is_supported (#1334) · 79e15ca9
      varunsh authored
      * Update is_supported
      * Return object from is_supported
      * Return by reference in interator
      79e15ca9
  25. 19 Aug, 2022 2 commits
  26. 18 Aug, 2022 1 commit
    • shivadbhavsar's avatar
      pybind updates for torch_migraphx library (#1323) · 8045f7c8
      shivadbhavsar authored
      Add function argument_from_pointer to allow directly passing a migraphx.shape object and a memory address. 
      Expose the is_compiled() method from migraphx::program. 
      Expose the enum types under migraphx::op. 
      8045f7c8
  27. 17 Aug, 2022 2 commits