File tree 3 files changed +1
-26
lines changed
torch/csrc/jit/codegen/cuda 3 files changed +1
-26
lines changed Original file line number Diff line number Diff line change @@ -298,21 +298,6 @@ void parallelizeAllLike(
298
298
}
299
299
}
300
300
301
- void computeAtInputs (TensorView* consumer, int pos, ComputeAtMode mode) {
302
- for (auto inp_tv : ir_utils::inputTvsOf (consumer)) {
303
- inp_tv->computeAt (consumer, pos, mode);
304
- }
305
- }
306
-
307
- void computeWithOutputs (TensorView* producer, int pos, ComputeAtMode mode) {
308
- for (auto out_tv : ir_utils::outputTvsOf (producer)) {
309
- if (out_tv == producer) {
310
- continue ;
311
- }
312
- producer->computeWith (out_tv, pos, mode);
313
- }
314
- }
315
-
316
301
namespace {
317
302
318
303
// Find the resolution points of the persistent buffers in the provided
Original file line number Diff line number Diff line change @@ -115,16 +115,6 @@ TORCH_CUDA_CU_API inline void parallelizeAllLike(
115
115
propagate_padding);
116
116
}
117
117
118
- TORCH_CUDA_CU_API void computeAtInputs (
119
- TensorView* consumer,
120
- int pos,
121
- ComputeAtMode mode = ComputeAtMode::Standard);
122
-
123
- TORCH_CUDA_CU_API void computeWithOutputs (
124
- TensorView* producer,
125
- int pos,
126
- ComputeAtMode mode = ComputeAtMode::Standard);
127
-
128
118
struct PersistentBufferInfo {
129
119
std::vector<TensorView*> persistent_buffers;
130
120
std::unordered_set<IterDomain*> unmappable_dims;
Original file line number Diff line number Diff line change @@ -1205,7 +1205,7 @@ TEST_F(NVFuserTest, FusionViewIdGraph_CUDA) {
1205
1205
fusion.addOutput (t13);
1206
1206
1207
1207
// Grab the trivial reduced tensor from t12's view.
1208
- auto tv11 = ir_utils::producerTvsOf ({ tv12} )[0 ];
1208
+ auto tv11 = ir_utils::producerTvsOf (tv12)[0 ];
1209
1209
1210
1210
// Start from the exact iter domain graph of the fusion
1211
1211
IterDomainGraph id_graph (&fusion);
You can’t perform that action at this time.
0 commit comments