├── .gitignore ├── LICENSE ├── README.md ├── actnn ├── actnn │ ├── __init__.py │ ├── _utils │ │ ├── __init__.py │ │ ├── collate.py │ │ ├── fetch.py │ │ ├── pin_memory.py │ │ ├── signal_handling.py │ │ └── worker.py │ ├── conf.py │ ├── cpp_extension │ │ ├── backward_func.cc │ │ ├── calc_precision.cc │ │ ├── ext_common.h │ │ ├── minimax.cc │ │ ├── minimax_cuda_kernel.cu │ │ ├── quantization.cc │ │ └── quantization_cuda_kernel.cu │ ├── dataloader.py │ ├── layers.py │ ├── module.py │ ├── ops.py │ ├── qbnscheme.py │ ├── qscheme.py │ └── utils.py └── setup.py ├── image_classification ├── Dockerfile ├── INSTALL.md ├── LICENSE ├── README.md ├── __init__.py ├── dist-test ├── dist-train ├── exp_mem_speed.py ├── image_classification │ ├── __init__.py │ ├── dataloaders.py │ ├── debug.py │ ├── logger.py │ ├── mixup.py │ ├── plot_variance.py │ ├── preact_resnet.py │ ├── resnet.py │ ├── smoothing.py │ ├── training.py │ └── utils.py ├── img │ └── .gitkeep ├── main.py ├── multiproc.py ├── quick_test.sh ├── run.sh ├── test ├── test_cifar ├── train └── train_cifar ├── mem_speed_benchmark ├── README.md ├── exp_mem_speed.py ├── mem_speed_r50.png ├── scaled_resnet.py └── train.py └── tests ├── dcgan_tutorial.py ├── test_act_quantized_ops.py ├── test_backward_func.py ├── test_conv_layer.py ├── test_linear.py ├── test_minimax.py ├── timeit_v2.py └── trigger_error.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ucbrise/actnn/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ucbrise/actnn/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ucbrise/actnn/HEAD/README.md -------------------------------------------------------------------------------- /actnn/actnn/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ucbrise/actnn/HEAD/actnn/actnn/__init__.py -------------------------------------------------------------------------------- /actnn/actnn/_utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ucbrise/actnn/HEAD/actnn/actnn/_utils/__init__.py -------------------------------------------------------------------------------- /actnn/actnn/_utils/collate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ucbrise/actnn/HEAD/actnn/actnn/_utils/collate.py -------------------------------------------------------------------------------- /actnn/actnn/_utils/fetch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ucbrise/actnn/HEAD/actnn/actnn/_utils/fetch.py -------------------------------------------------------------------------------- /actnn/actnn/_utils/pin_memory.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ucbrise/actnn/HEAD/actnn/actnn/_utils/pin_memory.py -------------------------------------------------------------------------------- /actnn/actnn/_utils/signal_handling.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ucbrise/actnn/HEAD/actnn/actnn/_utils/signal_handling.py -------------------------------------------------------------------------------- /actnn/actnn/_utils/worker.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ucbrise/actnn/HEAD/actnn/actnn/_utils/worker.py -------------------------------------------------------------------------------- /actnn/actnn/conf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ucbrise/actnn/HEAD/actnn/actnn/conf.py -------------------------------------------------------------------------------- /actnn/actnn/cpp_extension/backward_func.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ucbrise/actnn/HEAD/actnn/actnn/cpp_extension/backward_func.cc -------------------------------------------------------------------------------- /actnn/actnn/cpp_extension/calc_precision.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ucbrise/actnn/HEAD/actnn/actnn/cpp_extension/calc_precision.cc -------------------------------------------------------------------------------- /actnn/actnn/cpp_extension/ext_common.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ucbrise/actnn/HEAD/actnn/actnn/cpp_extension/ext_common.h -------------------------------------------------------------------------------- /actnn/actnn/cpp_extension/minimax.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ucbrise/actnn/HEAD/actnn/actnn/cpp_extension/minimax.cc -------------------------------------------------------------------------------- /actnn/actnn/cpp_extension/minimax_cuda_kernel.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ucbrise/actnn/HEAD/actnn/actnn/cpp_extension/minimax_cuda_kernel.cu -------------------------------------------------------------------------------- /actnn/actnn/cpp_extension/quantization.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ucbrise/actnn/HEAD/actnn/actnn/cpp_extension/quantization.cc -------------------------------------------------------------------------------- /actnn/actnn/cpp_extension/quantization_cuda_kernel.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ucbrise/actnn/HEAD/actnn/actnn/cpp_extension/quantization_cuda_kernel.cu -------------------------------------------------------------------------------- /actnn/actnn/dataloader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ucbrise/actnn/HEAD/actnn/actnn/dataloader.py -------------------------------------------------------------------------------- /actnn/actnn/layers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ucbrise/actnn/HEAD/actnn/actnn/layers.py -------------------------------------------------------------------------------- /actnn/actnn/module.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ucbrise/actnn/HEAD/actnn/actnn/module.py -------------------------------------------------------------------------------- /actnn/actnn/ops.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ucbrise/actnn/HEAD/actnn/actnn/ops.py -------------------------------------------------------------------------------- /actnn/actnn/qbnscheme.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ucbrise/actnn/HEAD/actnn/actnn/qbnscheme.py -------------------------------------------------------------------------------- /actnn/actnn/qscheme.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ucbrise/actnn/HEAD/actnn/actnn/qscheme.py -------------------------------------------------------------------------------- /actnn/actnn/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ucbrise/actnn/HEAD/actnn/actnn/utils.py -------------------------------------------------------------------------------- /actnn/setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ucbrise/actnn/HEAD/actnn/setup.py -------------------------------------------------------------------------------- /image_classification/Dockerfile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ucbrise/actnn/HEAD/image_classification/Dockerfile -------------------------------------------------------------------------------- /image_classification/INSTALL.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ucbrise/actnn/HEAD/image_classification/INSTALL.md -------------------------------------------------------------------------------- /image_classification/LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ucbrise/actnn/HEAD/image_classification/LICENSE -------------------------------------------------------------------------------- /image_classification/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ucbrise/actnn/HEAD/image_classification/README.md -------------------------------------------------------------------------------- /image_classification/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /image_classification/dist-test: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ucbrise/actnn/HEAD/image_classification/dist-test -------------------------------------------------------------------------------- /image_classification/dist-train: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ucbrise/actnn/HEAD/image_classification/dist-train -------------------------------------------------------------------------------- /image_classification/exp_mem_speed.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ucbrise/actnn/HEAD/image_classification/exp_mem_speed.py -------------------------------------------------------------------------------- /image_classification/image_classification/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ucbrise/actnn/HEAD/image_classification/image_classification/__init__.py -------------------------------------------------------------------------------- /image_classification/image_classification/dataloaders.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ucbrise/actnn/HEAD/image_classification/image_classification/dataloaders.py -------------------------------------------------------------------------------- /image_classification/image_classification/debug.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ucbrise/actnn/HEAD/image_classification/image_classification/debug.py -------------------------------------------------------------------------------- /image_classification/image_classification/logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ucbrise/actnn/HEAD/image_classification/image_classification/logger.py -------------------------------------------------------------------------------- /image_classification/image_classification/mixup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ucbrise/actnn/HEAD/image_classification/image_classification/mixup.py -------------------------------------------------------------------------------- /image_classification/image_classification/plot_variance.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ucbrise/actnn/HEAD/image_classification/image_classification/plot_variance.py -------------------------------------------------------------------------------- /image_classification/image_classification/preact_resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ucbrise/actnn/HEAD/image_classification/image_classification/preact_resnet.py -------------------------------------------------------------------------------- /image_classification/image_classification/resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ucbrise/actnn/HEAD/image_classification/image_classification/resnet.py -------------------------------------------------------------------------------- /image_classification/image_classification/smoothing.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ucbrise/actnn/HEAD/image_classification/image_classification/smoothing.py -------------------------------------------------------------------------------- /image_classification/image_classification/training.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ucbrise/actnn/HEAD/image_classification/image_classification/training.py -------------------------------------------------------------------------------- /image_classification/image_classification/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ucbrise/actnn/HEAD/image_classification/image_classification/utils.py -------------------------------------------------------------------------------- /image_classification/img/.gitkeep: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /image_classification/main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ucbrise/actnn/HEAD/image_classification/main.py -------------------------------------------------------------------------------- /image_classification/multiproc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ucbrise/actnn/HEAD/image_classification/multiproc.py -------------------------------------------------------------------------------- /image_classification/quick_test.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ucbrise/actnn/HEAD/image_classification/quick_test.sh -------------------------------------------------------------------------------- /image_classification/run.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ucbrise/actnn/HEAD/image_classification/run.sh -------------------------------------------------------------------------------- /image_classification/test: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ucbrise/actnn/HEAD/image_classification/test -------------------------------------------------------------------------------- /image_classification/test_cifar: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ucbrise/actnn/HEAD/image_classification/test_cifar -------------------------------------------------------------------------------- /image_classification/train: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ucbrise/actnn/HEAD/image_classification/train -------------------------------------------------------------------------------- /image_classification/train_cifar: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ucbrise/actnn/HEAD/image_classification/train_cifar -------------------------------------------------------------------------------- /mem_speed_benchmark/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ucbrise/actnn/HEAD/mem_speed_benchmark/README.md -------------------------------------------------------------------------------- /mem_speed_benchmark/exp_mem_speed.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ucbrise/actnn/HEAD/mem_speed_benchmark/exp_mem_speed.py -------------------------------------------------------------------------------- /mem_speed_benchmark/mem_speed_r50.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ucbrise/actnn/HEAD/mem_speed_benchmark/mem_speed_r50.png -------------------------------------------------------------------------------- /mem_speed_benchmark/scaled_resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ucbrise/actnn/HEAD/mem_speed_benchmark/scaled_resnet.py -------------------------------------------------------------------------------- /mem_speed_benchmark/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ucbrise/actnn/HEAD/mem_speed_benchmark/train.py -------------------------------------------------------------------------------- /tests/dcgan_tutorial.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ucbrise/actnn/HEAD/tests/dcgan_tutorial.py -------------------------------------------------------------------------------- /tests/test_act_quantized_ops.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ucbrise/actnn/HEAD/tests/test_act_quantized_ops.py -------------------------------------------------------------------------------- /tests/test_backward_func.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ucbrise/actnn/HEAD/tests/test_backward_func.py -------------------------------------------------------------------------------- /tests/test_conv_layer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ucbrise/actnn/HEAD/tests/test_conv_layer.py -------------------------------------------------------------------------------- /tests/test_linear.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ucbrise/actnn/HEAD/tests/test_linear.py -------------------------------------------------------------------------------- /tests/test_minimax.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ucbrise/actnn/HEAD/tests/test_minimax.py -------------------------------------------------------------------------------- /tests/timeit_v2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ucbrise/actnn/HEAD/tests/timeit_v2.py -------------------------------------------------------------------------------- /tests/trigger_error.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ucbrise/actnn/HEAD/tests/trigger_error.py --------------------------------------------------------------------------------