fix lint

apache · May 14, 2021 · 78bc8b7 · 78bc8b7
1 parent 7baf726
commit 78bc8b7
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 6 deletions.
diff --git a/include/tvm/relay/attrs/random.h b/include/tvm/relay/attrs/random.h
@@ -43,7 +43,8 @@ struct UniformAttrs : public tvm::AttrsNode<UniformAttrs> {
 
   TVM_DECLARE_ATTRS(UniformAttrs, "relay.attrs.UniformAttrs") {
     TVM_ATTR_FIELD(out_shape).describe("Shape of random numbers to generate");
-    TVM_ATTR_FIELD(out_dtype).set_default(NullValue<DataType>())
+    TVM_ATTR_FIELD(out_dtype)
+        .set_default(NullValue<DataType>())
         .describe("Data type of the generated numbers");
   }
 };

diff --git a/python/tvm/topi/random/kernel.py b/python/tvm/topi/random/kernel.py
@@ -512,11 +512,8 @@ def uniform_scalar(bits):
         standard_uniform = bits.astype(out_dtype) / float(1 << nfraction)
         return standard_uniform
 
-    standard_uniform_values = tvm.te.compute(
-        out_shape, lambda *i : uniform_scalar(random_bits(*i)))
+    standard_uniform_values = tvm.te.compute(out_shape, lambda *i: uniform_scalar(random_bits(*i)))
 
-    uniform_values = tvm.topi.add(
-                        tvm.topi.multiply(standard_uniform_values, high - low),
-                        low)
+    uniform_values = tvm.topi.add(tvm.topi.multiply(standard_uniform_values, high - low), low)
 
     return uniform_values