Skip to content

Instantly share code, notes, and snippets.

View priyathamkat's full-sized avatar
🏠
Working from home

Priyatham Kattakinda priyathamkat

🏠
Working from home
View GitHub Profile
@priyathamkat
priyathamkat / mp_infer.py
Last active February 14, 2024 18:11
Multi GPU inference using `torch.multiprocessing`
import torch
import torch.multiprocessing as mp
from absl import app, flags
from torchvision.models import AlexNet
FLAGS = flags.FLAGS
flags.DEFINE_integer("num_processes", 2, "Number of subprocesses to use")
@priyathamkat
priyathamkat / extended_euclid.cpp
Created September 18, 2016 00:08
Extended Euclidean algorithm.
#include<tuple>
std::tuple<long long, long long, long long> _extendedEuclid(long long a, long long b, long long x, long long y, long long _x, long long _y) {
if (a % b == 0) {
return std::make_tuple(b, x, y);
} else {
long long q = a / b;
return _extendedEuclid(b, a % b, _x - q * x, _y - q * y, x, y);
}
}