Update SDPUtils.cpp

intel · Dec 23, 2024 · 1e8cd76 · 1e8cd76
1 parent faec693
commit 1e8cd76
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/src/ATen/native/transformers/SDPUtils.cpp b/src/ATen/native/transformers/SDPUtils.cpp
@@ -4,6 +4,8 @@
 
 namespace sdp {
 
+using c10::array_of;
+
 bool check_all_tensors_on_device(sdp_params const& params, bool debug) {
   // Check that all tensors are on the GPU device
   // This should be handled by the stub dispatch, but whe call