fixed failing checks issue

Author: deivanayakisankaralingam

python/tvm/relax/frontend/torch/fx_translator.py

@@ -108,7 +108,7 @@ def _prelu_module(self, node: fx.Node) -> relax.Var:
         module = self.named_modules[node.target]
         alpha_tensor = module.weight.numpy()
         alpha = relax.const(alpha_tensor, dtype="float32")
-        axis = 0 if len(x.struct_info.shape) == 1 else 1 # Extract Channel size
+        axis = 0 if len(x.struct_info.shape) == 1 else 1  # Extract Channel size
         return self.block_builder.emit(relax.op.nn.prelu(x, alpha, axis))
 
     def _softmax_module(self, node: fx.Node) -> relax.Var:

src/relax/op/nn/nn.cc

@@ -79,6 +79,7 @@ TVM_REGISTER_OP("relax.nn.softplus")
     .set_attrs_type<SoftplusAttrs>()
     .set_attr<FInferStructInfo>("FInferStructInfo",
                                 InferStructInfoUnaryArith</*require_float_dtype=*/true>)
+    .set_attr<Bool>("FPurity", Bool(true));
 
 /* relax.nn.prelu */
 TVM_REGISTER_NODE_TYPE(PReluAttrs);