mirror of
https://github.com/pytorch/pytorch.git
synced 2025-10-20 21:14:14 +08:00
[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:
committed by
PyTorch MergeBot
parent
d9fe1713c3
commit
f4c0ef95bc
@ -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 {
|
||||
|
@ -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 {
|
||||
|
@ -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"),
|
||||
|
Reference in New Issue
Block a user