{ "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", " |