Skip to content

Commit df3393a

Browse files
authored
Some cleanup (#1957)
1 parent 7d1d7c8 commit df3393a

File tree

3 files changed

+1
-26
lines changed

3 files changed

+1
-26
lines changed

torch/csrc/jit/codegen/cuda/scheduler/utils.cpp

Lines changed: 0 additions & 15 deletions
Original file line numberDiff line numberDiff line change
@@ -298,21 +298,6 @@ void parallelizeAllLike(
298298
}
299299
}
300300

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-
316301
namespace {
317302

318303
// Find the resolution points of the persistent buffers in the provided

torch/csrc/jit/codegen/cuda/scheduler/utils.h

Lines changed: 0 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -115,16 +115,6 @@ TORCH_CUDA_CU_API inline void parallelizeAllLike(
115115
propagate_padding);
116116
}
117117

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-
128118
struct PersistentBufferInfo {
129119
std::vector<TensorView*> persistent_buffers;
130120
std::unordered_set<IterDomain*> unmappable_dims;

torch/csrc/jit/codegen/cuda/test/test_gpu_view.cpp

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1205,7 +1205,7 @@ TEST_F(NVFuserTest, FusionViewIdGraph_CUDA) {
12051205
fusion.addOutput(t13);
12061206

12071207
// 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];
12091209

12101210
// Start from the exact iter domain graph of the fusion
12111211
IterDomainGraph id_graph(&fusion);

0 commit comments

Comments
 (0)