4 lines
118 B
Plaintext
4 lines
118 B
Plaintext
#pragma once
|
|
// TODO: Remove once torchvision has been updated to use the ATen header
|
|
#include <ATen/cuda/Atomic.cuh>
|