{ "cells": [ { "cell_type": "code", "execution_count": 77, "metadata": {}, "outputs": [], "source": [ "import pandas as pd\n", "\n", "df = pd.read_csv('allocs7.csv')\n", "# remove all views\n", "df = df[df['tensor_view_src_id'] == \"(nil)\"]\n", "df = df.drop_duplicates(subset=['tensor_id'])\n", "df = df.sort_values(by='size', ascending=False, kind='stable')" ] }, { "cell_type": "code", "execution_count": 78, "metadata": {}, "outputs": [ { "data": { "text/html": [ "
\n", " | buffer_id | \n", "buffer_name | \n", "offset+size | \n", "
---|---|---|---|
7 | \n", "0x56fa9b3b02f0 | \n", "CPU | \n", "430940160 | \n", "
1 | \n", "0x56fa95d0a890 | \n", "CUDA0 | \n", "402653184 | \n", "
5 | \n", "0x56fa9af97b00 | \n", "CUDA0 | \n", "4294705152 | \n", "
6 | \n", "0x56fa9afff860 | \n", "CUDA0 | \n", "155197440 | \n", "
0 | \n", "0x56fa95cd17d0 | \n", "CUDA1 | \n", "134217728 | \n", "
3 | \n", "0x56fa9ae68ec0 | \n", "CUDA1 | \n", "321404928 | \n", "
4 | \n", "0x56fa9af5b340 | \n", "CUDA1 | \n", "1862524928 | \n", "
2 | \n", "0x56fa9ae63710 | \n", "CUDA_Host | \n", "41967616 | \n", "