Jiaming Yuan
|
8189126d51
|
Add CUDA iterator to tensor view. (#10074)
|
2024-03-01 14:15:31 +08:00 |
|
Jiaming Yuan
|
42de9206fc
|
Support multi-target, fit intercept for hinge. (#9850)
|
2023-12-08 05:50:41 +08:00 |
|
Jiaming Yuan
|
8c676c889d
|
Remove internal use of gpu_id. (#9568)
|
2023-09-20 23:29:51 +08:00 |
|
Jiaming Yuan
|
ddf2e68821
|
Use the new DeviceOrd in the linalg module. (#9527)
|
2023-08-29 13:37:29 +08:00 |
|
Jiaming Yuan
|
04aff3af8e
|
Define the new device parameter. (#9362)
|
2023-07-13 19:30:25 +08: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 |
|