1 | #include <ATen/Dispatch.h> |
---|---|
2 | #include <ATen/record_function.h> |
3 | |
4 | namespace at::detail { |
5 | |
6 | void record_kernel_function_dtype(std::string name) { |
7 | RECORD_FUNCTION_WITH_SCOPE( |
8 | at::RecordScope::KERNEL_FUNCTION_DTYPE, |
9 | std::move(name), |
10 | c10::ArrayRef<const c10::IValue>{}); |
11 | } |
12 | |
13 | } // namespace at::detail |
14 |