sandeepkumar-skb

import onnx
import sys

name = sys.argv[1]
model = onnx.load(name)
onnx.checker.check_model(model)
print(onnx.helper.printable_graph(model.graph))

sandeepkumar-skb

import tensorrt as trt
import numpy as np
import pycuda.autoinit
import pycuda.driver as cuda 
import time

model_path = "model.onnx"
input_size = 32

TRT_LOGGER = trt.Logger(trt.Logger.WARNING)

sandeepkumar-skb

# Path to your oh-my-zsh installation.
export ZSH="$HOME/.oh-my-zsh"

# Set name of the theme to load --- if set to "random", it will
# load a random theme each time oh-my-zsh is loaded, in which case,
# to know which specific one was loaded, run: echo $RANDOM_THEME
# See https://github.com/ohmyzsh/ohmyzsh/wiki/Themes
#ZSH_THEME="agnoster"
########################
ZSH_THEME="powerlevel9k/powerlevel9k"

sandeepkumar-skb

#!/bin/bash

sudo apt-get update
sudo apt upgrade

sudo apt install zsh
sudo apt-get install powerline fonts fonts-powerline

echo "################Cloning OH MY ZSH####################"
echo ""

sandeepkumar-skb

" Use Vim settings, rather then Vi settings (much better!).
" This must be first, because it changes other options as a side effect.
set nocompatible

" TODO: this may not be in the correct place. It is intended to allow overriding <Leader>.
" source ~/.vimrc.before if it exists.
if filereadable(expand("~/.vimrc.before"))
  source ~/.vimrc.before
endif

sandeepkumar-skb

docker run -it --ipc=host --gpus all --user $(id -u):$(id -g) -v /etc/passwd:/etc/passwd:ro -v /etc/group:/etc/group:ro -v `pwd`:/host <docker_image>

sandeepkumar-skb

'''
Memory profiling utilities
'''
import gc
import inspect
import linecache
import os.path
import sys
import time
import threading

sandeepkumar-skb

'''
Memory profiling utilities
'''
import gc
import inspect
import linecache
import os.path
import sys
import time
import threading

sandeepkumar-skb

#include <iostream>
#include <chrono>

void cpu_all_reduce(int* sum, int* data, int n){
    int temp_sum = 0;
    for (int i=0; i<n; ++i){
        temp_sum += data[i];
    }
    *sum = temp_sum;
}

sandeepkumar-skb

#include <iostream>
#include <chrono>


void cpu_all_reduce(int* sum, int* data, int n){
    int temp_sum = 0;
    for (int i=0; i<n; ++i){
        temp_sum += data[i];
    }
    *sum = temp_sum;