ColossalAI/examples/tutorial/new_api
Hongxin Liu d202cc28c0
[npu] change device to accelerator api (#5239)
* update accelerator

* fix timer

* fix amp

* update

* fix

* update bug

* add error raise

* fix autocast

* fix set device

* remove doc accelerator

* update doc

* update doc

* update doc

* use nullcontext

* update cpu

* update null context

* change time limit for example

* udpate

* update

* update

* update

* [npu] polish accelerator code

---------

Co-authored-by: Xuanlei Zhao <xuanlei.zhao@gmail.com>
Co-authored-by: zxl <43881818+oahzxl@users.noreply.github.com>
2024-01-09 10:20:05 +08:00
..
cifar_resnet [npu] change device to accelerator api (#5239) 2024-01-09 10:20:05 +08:00
cifar_vit [npu] change device to accelerator api (#5239) 2024-01-09 10:20:05 +08:00
glue_bert [npu] change device to accelerator api (#5239) 2024-01-09 10:20:05 +08:00
README.md [booster] implemented the torch ddd + resnet example (#3232) 2023-03-27 10:24:14 +08:00
test_ci.sh [example] add train resnet/vit with booster example (#3694) 2023-05-08 10:42:30 +08:00

README.md

New API Features

The New API is not officially released yet.

This folder contains some of the demonstrations of the new API. The new API is still under intensive development and will be released soon.