Skip to content

Commit

Permalink
fix lint
Browse files Browse the repository at this point in the history
  • Loading branch information
merrymercy committed Jul 22, 2020
1 parent 56b0187 commit 34d4d7b
Show file tree
Hide file tree
Showing 2 changed files with 3 additions and 5 deletions.
6 changes: 2 additions & 4 deletions include/tvm/auto_scheduler/compute_dag.h
Original file line number Diff line number Diff line change
Expand Up @@ -117,8 +117,7 @@ class AccessAnalyzer : public ObjectRef {
* \note This function propagates the relation for inlined ops
*/
TVM_DLL void GetConsumers(
const State& state,
const te::Operation& op,
const State& state, const te::Operation& op,
std::unordered_set<te::Operation, ObjectHash, ObjectEqual>* consumers) const;

/*!
Expand All @@ -129,8 +128,7 @@ class AccessAnalyzer : public ObjectRef {
* \note This function propagates the relation for inlined ops
*/
TVM_DLL void GetProducers(
const State& state,
const te::Operation& op,
const State& state, const te::Operation& op,
std::unordered_set<te::Operation, ObjectHash, ObjectEqual>* producers) const;

/*!
Expand Down
2 changes: 1 addition & 1 deletion tests/python/unittest/test_auto_scheduler_common.py
Original file line number Diff line number Diff line change
Expand Up @@ -33,7 +33,7 @@ def matmul_auto_scheduler_test(N, M, K):


@auto_scheduler.register_workload("matmul_auto_scheduler_test_rename_1")
def matmul_auto_scheduler_test_rename_1(N, M, K):
def matmul_auto_scheduler_test_rename_0(N, M, K):
A = te.placeholder((N, K), name='A')
B = te.placeholder((K, M), name='B')
k = te.reduce_axis((0, K), name='k')
Expand Down

0 comments on commit 34d4d7b

Please sign in to comment.