[AMD] Fix broken build from nested transformer utils (#110245)

Summary: D49374910 breaks internal amd build because we didn't hipify the header file in nested/cuda. Maybe it's just easier to move it outside.

Reviewed By: nmacchioni

Differential Revision: D49743234

Pull Request resolved: https://github.com/pytorch/pytorch/pull/110245
Approved by: https://github.com/drisspg
This commit is contained in:
Xiaodong Wang
2023-10-03 08:05:10 +00:00
committed by PyTorch MergeBot
parent d9fe1713c3
commit f4c0ef95bc
4 changed files with 2 additions and 3 deletions

View File

@ -15,10 +15,10 @@
#include <ATen/native/NonSymbolicBC.h>
#include <ATen/native/nested/NestedTensorTransformerFunctions.h>
#include <ATen/native/nested/NestedTensorTransformerUtils.h>
#include <ATen/native/nested/NestedTensorMath.h>
#include <ATen/native/nested/NestedTensorUtils.h>
#include <ATen/native/transformers/cuda/sdp_utils.h>
#include <ATen/native/nested/cuda/NestedTensorTransformerUtils.h>
#include <ATen/cuda/CUDAContext.h>
namespace at {

View File

@ -1,6 +1,6 @@
#include <ATen/ATen.h>
#include <ATen/NestedTensorImpl.h>
#include <ATen/native/nested/cuda/NestedTensorTransformerUtils.h>
#include <ATen/native/nested/NestedTensorTransformerUtils.h>
#include <iostream>
#include <tuple>
namespace at {

View File

@ -842,7 +842,6 @@ def define_buck_targets(
("aten/src", "ATen/native/cpu/**/*.h"),
("aten/src", "ATen/native/sparse/*.h"),
("aten/src", "ATen/native/nested/*.h"),
("aten/src", "ATen/native/nested/cuda/*.h"),
("aten/src", "ATen/native/quantized/*.h"),
("aten/src", "ATen/native/quantized/cpu/*.h"),
("aten/src", "ATen/native/transformers/*.h"),