Commit 574864d8 authored by umangyadav's avatar umangyadav
Browse files

add trace_eval=3

parent e4ef64f4
......@@ -21,6 +21,7 @@
* OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
* THE SOFTWARE.
*/
#include "migraphx/file_buffer.hpp"
#include <migraphx/version.h>
#include <migraphx/compile_options.hpp>
#include <migraphx/program.hpp>
......@@ -584,6 +585,23 @@ std::vector<argument> program::eval(parameter_map params, execution_environment
std::cout << std::endl;
print_statistics(std::cout, buffer);
}
else if(trace_level == 3)
{
auto buffer_s = buffer.get_shape();
if(buffer_s.type() != shape::tuple_type)
{
auto std_buffer = buffer;
if(not buffer_s.standard())
{
std_buffer =
migraphx::make_op("contiguous")
.compute(shape{buffer_s.type(), buffer_s.lens()}, {buffer});
}
std::vector<char> binary_vec(
std_buffer.data(), std_buffer.data() + std_buffer.get_shape().bytes());
write_buffer(ins_out.at(ins) + ".bin", binary_vec);
}
}
else
{
std::cout << "Output: " << buffer << std::endl;
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment