Skip to content

Commit a83a1bc

Browse files
anthonyalayopytorchmergebot
authored andcommitted
Adding c10 device type to newly added DeviceAccelerator (pytorch#119961)
Follow up to pytorch#104364, A new file got submitted yesterday that is using DeviceType without the c10 namespace. This fixes that. I haven't yet figured out a way to setup a test for this, but I will submit a follow up PR once I figure that out. Pull Request resolved: pytorch#119961 Approved by: https://github.com/ezyang
1 parent e5bfdde commit a83a1bc

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

aten/src/ATen/DeviceAccelerator.h

+1-1
Original file line numberDiff line numberDiff line change
@@ -22,6 +22,6 @@ namespace at {
2222
// Ensures that only one accelerator is available (at
2323
// compile time if possible) and return it.
2424
// When checked is true, the returned optional always has a value.
25-
TORCH_API std::optional<DeviceType> getAccelerator(bool checked = false);
25+
TORCH_API std::optional<c10::DeviceType> getAccelerator(bool checked = false);
2626

2727
} // namespace at

0 commit comments

Comments
 (0)