Skip to content

GitLab

Switch branch/tag
Find file Normal viewHistoryPermalink
autocast.h 97 Bytes
Newer Older
mcarilli's avatar
[WIP] Allow autocast for 1.6 (#2384)  
mcarilli committed
1
2 
#pragma once
Ashish Farmer's avatar
Enable autocast for NMS and ROIAlign on ROCm (#2637)  
Ashish Farmer committed
3 
#if defined(WITH_CUDA) || defined(WITH_HIP)
mcarilli's avatar
compiles (#2646)  
mcarilli committed
4 
#include <ATen/autocast_mode.h>
mcarilli's avatar
[WIP] Allow autocast for 1.6 (#2384)  
mcarilli committed
5 
#endif