File size: 118 Bytes
52d77e3 | 1 2 3 4 | #pragma once
// TODO: Remove once torchvision has been updated to use the ATen header
#include <ATen/cuda/Atomic.cuh>
|
52d77e3 | 1 2 3 4 | #pragma once
// TODO: Remove once torchvision has been updated to use the ATen header
#include <ATen/cuda/Atomic.cuh>
|