Enable CUDA target in TVM CLI

autowarefoundation / modelzoo

A collection of machine-learned models for use in autonomous driving applications.

https://www.autoware.org/

Apache License 2.0

62 stars 24 forks source link

Closed LucaFos closed 3 years ago

LucaFos commented 3 years ago

Dockerfile for TVM+CUDA provided. Added TVM CLI options to provide compilation for CUDA target.

Change-Id: I0b39c36bf4842a674c54ac30b3e443d395af4c6e Signed-off-by: Luca Foschiani luca.foschiani@arm.com Issue-Id: SCM-1391