Граф коммитов

9 Коммитов

Автор SHA1 Сообщение Дата
Ritwik Das 131945043d remove dependency from bfloat16 2022-05-26 20:06:24 -07:00
Ritwik Das 7d86c17344 Add bfloat16 support in hatlib 2022-05-26 19:19:14 -07:00
Mason Remy a23a2be45b Fix array volume computation for arrays larger than 2-D
We've been taking a product of strides and multiplying that by the major
dimension, but we should be just taking the largest stride and
multiplying that by the major dimension, as the largest stride
already factors in the other strides in the array shape.
2022-05-10 18:37:20 -07:00
Mason Remy a50a524f57 Fix byte size calculation to support layouts other than FIRST_MAJOR 2022-05-06 13:00:18 -07:00
Kern Handa 5367112387
Adds support for verifying/benchmarking functions with layouts other than first major (#43)
* Adds support for function parameters with layouts other than first major
2022-04-14 22:15:08 -07:00
Lisa Ong e03bc274fa
Support unsigned int types in HAT file (#37)
* added unsigned int types

* updated readme

* PR feedback

* [nfc] order entries by size

* typo

* [nfc] restored comment
2022-03-31 15:48:19 +08:00
Kern Handa 544fcc0e01
Refactors and restructures code for tighter integration of HAT object model (#36)
Changes:
* utility functions in `hatlib` now make use of the HAT object model (OM). Previously, `tomlkit` was being using directly and was bypassing the checks and OM construction of child objects.
* `hatlib.load` now returns a `HATPackage` object that results from deserializing the HAT file passed in. Additionally, if functions can be loaded, a non-empty `AttributeDict` is returned as a second value.
* Moved test folder to be a sibling of library folder. Now the package no longer contains the tests when installed. Additionally, the library can be tested by running `python -m unittest discover test` in the root of the repository.
* Adds YAPF style config to setup.cfg and formatted all touched files with `yapf`
2022-03-30 13:50:00 -07:00
Kern Handa f0e70a49cd
Adds ROCm support for device functions (#33)
Also adds the helper command `hatlib.verify_hat`

```
usage: hatlib.verify_hat [-h] hat_path

Executes every available function in the hat package with randomized inputs. Meant for quick verification. Example: hatlib.verify_hat_package <hat_path>

positional arguments:
  hat_path    Path to the HAT file

optional arguments:
  -h, --help  show this help message and exit
```
2022-03-20 21:17:56 -07:00
Kern Handa 615006cecb
Add beginnings of support for CUDA device functions (#32)
Adds support for GPU, device functions, and launch functions
2022-03-14 23:20:41 -07:00