malfet’s gists

malfet / gh_association_curiosities.py

Created January 26, 2022 17:07

Querying author association from github

	#!/usr/bin/env python3

	# Author of https://github.com/pytorch/pytorch/pull/71735 is me (malfet)
	# And if executed with my PAT it will return my association as "MEMBER"
	# But if executed by somebody outside of org, it will say that association is "CONTRIBUTOR"
	# Instead of getting PAT, one can simply execute the query at https://docs.github.com/en/graphql/overview/explorer
	GH_GET_ASSOCIATION = """
	query {
	repository(owner: "pytorch", name: "pytorch") {
	pullRequest(number: 71735) {

malfet / get_default_branch.py

Created January 19, 2022 17:32

Print default branches for public repos in pytorch org

	#!/usr/bin/env python3
	# Print default branch for public repos in pytorch org
	# Requires github token
	from typing import Any, Dict

	GH_SEARCH_REPO_REQ = """
	query($cursor: String) {
	search(type: REPOSITORY, query: "org:pytorch", first: 100, after: $cursor) {
	nodes {
	... on Repository {

malfet / GH patch and diff are not the same

Created January 7, 2022 20:17

	% curl -L https://github.com/pytorch/pytorch/pull/69840.patch\|diffstat
	% Total % Received % Xferd Average Speed Time Time Time Current
	Dload Upload Total Spent Left Speed
	100 143 100 143 0 0 459 0 --:--:-- --:--:-- --:--:-- 468
	100 7148 0 7148 0 0 11172 0 --:--:-- --:--:-- --:--:-- 11172
	aten/src/ATen/templates/RegisterBackendSelect.cpp \| 27 ++++++++++++++-------------
	aten/src/ATen/templates/RegisterSchema.cpp \| 16 +---------------
	tools/codegen/gen.py \| 40 +++++++++++++++++++++++-----------------
	3 files changed, 38 insertions(+), 45 deletions(-)
	% curl -L https://github.com/pytorch/pytorch/pull/69840.diff\|diffstat

malfet / torch_query_memory.py

Created November 24, 2021 01:46

Query PyTorch memory utilisation

	#!/usr/bin/env python3

	import os
	from subprocess import check_output

	def get_pmem():
	pmem = check_output(["sh", "-c", f"pmap -d {os.getpid()}\|tail -n 1"], encoding="latin1").split()[3]
	return pmem

	def get_gpumem():

malfet / hello.cu

Last active November 24, 2021 21:05

CUDA Hello world

	#include <stdio.h>
	__global__ void kernel() {
	printf("Hello World of CUDA %d\n", threadIdx.x);
	}

	int main() {
	kernel<<<1,1>>>();
	return cudaDeviceSynchronize();
	}

malfet / gist:175d76815eebf8b2f19005cb80105b45

Created November 1, 2021 17:44

pytorch.github.io broken links

	URL `resources/fonts/dejavu.css'
	Parent URL file:///home/nshulga/git/pytorch.github.io/javadoc/stylesheet.css, line 3, col 12
	Real URL file:///home/nshulga/git/pytorch.github.io/javadoc/resources/fonts/dejavu.css
	Result Error: URLError: <urlopen error [Errno 2] No such file or directory: '/home/nshulga/git/pytorch.github.io/javadoc/resources/fonts/dejavu.css'>

	URL `../fonts/FreightSans/freight-sans-bold.woff'
	Parent URL file:///home/nshulga/git/pytorch.github.io/assets/main.css, line 1, col 147264
	Real URL file:///home/nshulga/git/pytorch.github.io/fonts/FreightSans/freight-sans-bold.woff
	Result Error: URLError: <urlopen error [Errno 2] No such file or directory: '/home/nshulga/git/pytorch.github.io/fonts/FreightSans/freight-sans-bold.woff'>

malfet / whatsmyname.cpp

Created October 7, 2021 16:41

Prints name of the template argument

	#include <iostream>
	#include <string>

	template<typename T>
	struct WhatsMyName {
	WhatsMyName() {
	std::cout << __PRETTY_FUNCTION__ << std::endl;
	}
	};

malfet / mup-smaller.cpp

Created September 28, 2021 13:49

clang-13 crash using omp critical pragma

	// c++ -std=c++14 -O2 -fPIC -Xpreprocessor -fopenmp mup-smaller.cpp
	//
	#include <omp.h>
	#include <atomic>
	#include <exception>
	#include <algorithm>

	#define C10_UNLIKELY(expr) (__builtin_expect(static_cast<bool>(expr), 0))

	namespace at {

malfet / conda_remove_abi.py

Created August 5, 2021 04:46

Ugly script that removes python_abi dep from conda packages

	#!/usr/bin/env python3
	from conda_build.convert import extract_temporary_directory, create_target_archive
	import shutil
	import os
	import sys

	def remove_line_with_pattern(fname, pattern):
	with open(fname) as f:
	lines = f.read().split('\n')
	del_idx = []

malfet / neon-trunc.cpp

Created June 11, 2021 14:23

Implements trunc using vcvtq vs vrndq

	#include <arm_neon.h>
	#include <iostream>

	int main(void) {
	const float inp[4] = { 3.1, 2.7, -2.9, -1.0000000200408773e+20 };
	std::cout << "inp= " << inp[0] << " " << inp[1] << " " << inp[2] << " " << inp[3] << std::endl;
	float32x4_t f = vld1q_f32(inp);
	float32x4_t old_trunc = vcvtq_f32_s32(vcvtq_s32_f32(f));
	float32x4_t new_trunc = vrndq_f32(f);
	float out[4];

Nikita Shulga malfet