Hui Liu
|
82d81bca94
|
rm hip.h files
|
2023-10-30 21:54:00 -07:00 |
|
Hui Liu
|
782b73f2bb
|
rocm enable for v2.0.1
|
2023-10-27 18:50:28 -07:00 |
|
Jiaming Yuan
|
04aff3af8e
|
Define the new device parameter. (#9362)
|
2023-07-13 19:30:25 +08:00 |
|
amdsc21
|
b22644fc10
|
add hip.h
|
2023-05-20 01:25:33 +02:00 |
|
Jiaming Yuan
|
f236640427
|
Support F order for the tensor type. (#8872)
- Add F order support for tensor and view.
- Use parameter pack for automatic type cast. (avoid excessive static cast for shape).
|
2023-03-08 03:27:49 +08:00 |
|
Jiaming Yuan
|
4d665b3fb0
|
Restore clang tidy test. (#8861)
|
2023-03-03 13:47:04 -08:00 |
|
Jiaming Yuan
|
3e26107a9c
|
Rename and extract Context. (#8528)
* Rename `GenericParameter` to `Context`.
* Rename header file to reflect the change.
* Rename all references.
|
2022-12-07 04:58:54 +08:00 |
|
Jiaming Yuan
|
98d6faefd6
|
Implement slope for Pseduo-Huber. (#7727)
* Add objective and metric.
* Some refactoring for CPU/GPU dispatching using linalg module.
|
2022-03-14 21:42:38 +08:00 |
|
Jiaming Yuan
|
85cbd32c5a
|
Add range-based slicing to tensor view. (#7453)
|
2021-11-27 13:42:36 +08:00 |
|
Jiaming Yuan
|
a7057fa64c
|
Implement typed storage for tensor. (#7429)
* Add `Tensor` class.
* Add elementwise kernel for CPU and GPU.
* Add unravel index.
* Move some computation to compile time.
|
2021-11-14 18:53:13 +08:00 |
|