text: // ***************************************************************
  // Copyright (c) 2020 Jittor. Authors: Dun Liang <randonlang@gmail.com>. All Rights Reserved.
  // This file is subject to the terms and conditions defined in
  // file 'LICENSE.txt', which is part of this source code package.
  // ***************************************************************
  #pragma once
  #include "op.h"
  
  namespace jittor {
  
  struct CublasTestOp : Op {
      Var* output;
      int size_mult;
  
      CublasTestOp(int size_mult);
      
      const char* name() const override { return "cublas_test"; }
      DECLARE_jit_run;
  };
  
  } // jittor